avgpool的bug

[WeiMinghong0517]

我用hrsc训练的时候，通道注意力的

def forward(self, x): avg_pool = F.avg_pool2d(x, x.size(2), stride=x.size(2))# 平均池化操作 res=self.gate_c(avg_pool).unsqueeze(2).unsqueeze(3).expand_as(x)# 通过通道注意力模块处理平均池化的结果 return res 平均池化报错了，RuntimeError: Given input size: (256x100x96). Calculated output size: (256x1x0). Output size is too small 因该是部分图片大小不太规矩，请问您当时是怎么解决的呢？

[IsenLwww]

我也遇到这个问题了，请问您当时解决了吗

[WeiMinghong0517]

我也遇到这个问题了，请问您当时解决了吗

因为数据集有一张图片长宽是反的 h<w，所以你需要加一个判断，在 ’tride = x.size(2) if x.size(2) > x.size(3): stride = x.size(3) avg_pool = F.avg_pool2d(x, stride, stride=stride) # 平均池化操作 res = self.gate_c(avg_pool).unsqueeze(2).unsqueeze(3).expand_as(x)

给哥们点个赞

[IsenLwww]

我也遇到这个问题了，请问您当时解决了吗

因为数据集有一张图片长宽是反的 h<w，所以你需要加一个判断，在 ’tride = x.size(2) if x.size(2) > x.size(3): stride = x.size(3) avg_pool = F.avg_pool2d(x, stride, stride=stride) # 平均池化操作 res = self.gate_c(avg_pool).unsqueeze(2).unsqueeze(3).expand_as(x)

给哥们点个赞

谢谢大哥，太牛了

[IsenLwww]

我也遇到这个问题了，请问您当时解决了吗

因为数据集有一张图片长宽是反的 h<w，所以你需要加一个判断，在 ’tride = x.size(2) if x.size(2) > x.size(3): stride = x.size(3) avg_pool = F.avg_pool2d(x, stride, stride=stride) # 平均池化操作 res = self.gate_c(avg_pool).unsqueeze(2).unsqueeze(3).expand_as(x)

给哥们点个赞

大哥请问你遇到过这个问题吗 在losses.py里 max_pos, armmax_pos = pos.max(0) RuntimeError: cannot perform reduction function max on tensor with no elements because the operation does not have an identity 自己解决得好像有些问题