8个计算机视觉深度学习中常见的Bug

 a, b, c, d = i, min(h, i + self.size), j, min(w, j + self.size) 

 patch = x[a:b, c:d, :] 

 mask[a:b, c:d, :] += np.expand_dims(self.predictor(patch), -1) 

 weights[a:b, c:d, :] = 1 

 return mask / weights 

有一个符号输入错误，代码段足够大，可以很容易地找到它。我怀疑仅仅通过代码就能快速识别它。但是很容易检查代码是否正确：

class Model(nn.Module): 

 def forward(self, x): 

 return x.mean(axis=-1) 

model = Model() 

grid_predictor = GridPredictor(model, size=128, stride=64) 

simple_pred = np.expand_dims(model(img), -1)  

grid_pred = grid_predictor(img) 

np.testing.assert_allclose(simple_pred, grid_pred, atol=.001) 

--------------------------------------------------------------------------- 

AssertionError Traceback (most recent call last) 

<ipython-input-24-a72034c717e9> in <module> 

 9 grid_pred = grid_predictor(img) 

 10  

---> 11 np.testing.assert_allclose(simple_pred, grid_pred, atol=.001) 

（编辑：晋中站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!