解决YOLOv5环境AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘

运行detect.py时会出现如下报错

解决YOLOv5环境AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘_第1张图片

根据错误提示找到unsampling.py文件的第153行和第154行,如图所示

解决YOLOv5环境AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘_第2张图片

删除第154行及153行的都好,保留如图所示即可

解决YOLOv5环境AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘_第3张图片

再次运行就没问题了,完美解决

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