错误一:
AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
解决方法,修改pytorch 源码:
Lib\site-packages\torch\nn\modules下找到文件upsampling.py
def forward(self, input: Tensor) -> Tensor:
# return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners,
# recompute_scale_factor=self.recompute_scale_factor)
return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)
错误二:
RuntimeError: result type Float can't be cast to the desired output type xxx
yolov5下的 【utils 文件夹】中的【loss.py】
需要修改的两处(注释的代码时之前的代码,下面的是修改的):
修改的两处类容如下:
anchors, shape = self.anchors[i], p[i].shape # 修改,2022年7月11日
indices.append((b, a, gj.clamp_(0, shape[2] - 1), gi.clamp_(0, shape[3] - 1))) # image, anchor, grid # 修改,2022年7月11日