前言:使用yolov5-6.0版本想检测几张图,就出现了题目中的问题,可以看出是上采样函数不太对,现把解决方案记录下。
版本:Yolov5-6.0、Python3.8、Pytorch1.11.0
一、问题复现
二、官网解决方案
这个问题在yolov5版本中第一次出现,而且是和Pytorch 1.11.0有关系。
就是说,在train和detect时都有可能会遇到这个问题 。下面是降低Pytorch版本解决方案,降到10以内。
然后这个博主又针对Pytorch版本1.11.0的做了fix
但是好像没解决,笔者torch版本是1.11.0,还是出现这个问题了。
网友给出的修改上采样函数解决方案。
确实解决了,网友666。
三、引用文献
3.1 PyTorch 1.11.0 compatibility updates (#1914) · ultralytics/yolov3@7093a2b · GitHub
3.2 https://github.com/ultralytics/yolov5/issues/5499
3.3 AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘_阳光不锈@的博客-CSDN博客