YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效

目录

  • 报错:AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor'
  • 解决方法
  • 问题解决
  • 注意事项

报错:AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

如图:
YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第1张图片

解决方法

1.点击报错行该路径,进入编辑页
YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第2张图片


2.将原代码(153-154行)修改为如下所示(155行):
YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第3张图片
即:

return F.interpolate(input, self.size, self.scale_factor, self.mode, self.align_corners)

问题解决

保存后再次运行detect.py,即可:
YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第4张图片


来看看训练结果吧
YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第5张图片

YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第6张图片

注意事项

运行detect.py之前,应该将YOLOV5对应的权重文件放到项目文件夹下:
YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_第7张图片
这里提供官方权重文件,可点击下方链接自行下载:

YOLOv5权重文件

https://download.csdn.net/download/weixin_43401024/85225886

你可能感兴趣的:(python,深度学习,pytorch,pycharm)