pytorch gpu版 yolov5 报错
错误一:
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日
作者:宋桓公
出处:http://www.cnblogs.com/douzi2/
如果您觉得阅读本文对您有帮助,请点一下“推荐”按钮,您的“推荐”将是我最大的写作动力!欢迎各位转载,但是未经作者本人同意,转载文章之后必须在文章页面明显位置给出作者和原文连接,否则保留追究法律责任的权利。