YOLOv5报AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

最新推荐文章于 2023-12-13 11:41:38 发布

原创

最新推荐文章于 2023-12-13 11:41:38 发布 · 1.1w 阅读

23 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #pytorch #python

在运行YOLOv5时遇到了AttributeError，具体为'Upsample'对象没有'recompute_scale_factor'属性。错误发生在torch版本1.11.0和torchvision版本0.10.2下。解决方法是将torchvision版本降低到0.10.1，并将torch版本降低到1.9.1。

Traceback (most recent call last):
File “E:\yolo\yolov5-5.0\yolov5-5.0\detect.py”, line 72, in detect
pred = model(img, augment=opt.augment)[0]
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\module.py”, line 1110, in _call_impl
return forward_call(*input, **kwargs)
File “E:\yolo\yolov5-5.0\yolov5-5.0\models\yolo.py”, line 123, in forward
return self.forward_once(x, profile) # single-scale inference, train
File “E:\yolo\yolov5-5.0\yolov5-5.0\models\yolo.py”, line 139, in forward_once
x = m(x) # run
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\module.py”, line 1110, in _call_impl
return forward_call(*input, **kwargs)
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\upsampling.py”, line 154, in forward
recompute_scale_factor=self.recompute_scale_factor)
File “D:\python\anaconda\lib\site-packages\torch\nn\modules\module.py”, line 1185, in getattr
raise AttributeError("’{}’ obj