💪 专业从事且热爱图像处理,图像处理专栏更新如下👇:
📝《图像去噪》
📝《超分辨率重建》
📝《语义分割》
📝《风格迁移》
📝《目标检测》
📝《图像增强》
📝《模型优化》
📝《模型实战部署》
📝《图像配准融合》
📝《数据集》
📝《高效助手》
一、问题
在使用YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample’ object has no attribute 'recompute_scale_factor’的快速解决方法。
二、解决办法
2.1 解决方法一
注意下面包的版本号:
参考其它博主说降低torchhe和torchvision版本,比如上图所示我的torch版本1.11.0 torchvision版本0.10.2,torch版本降低到版本1.9.1,torchvision版本降低到版本0.10.1。这是一种解决办法,但是要重新pytorch,我就嫌很麻烦,配置安装过程中可能又出现各种问题,所以我没有重新安装,采用了下面的这种方法,不用降低版本重载,就可以快速解决问题,见下。
2.2 解决方法二
2.2.1 问题分析
我们先看一下问题:
这个问题只要顺着路径找到文件upsampling.py并打开,修改里面的代码即可解决。
2.2.2 解决方法
找到下面框选出来的脚本打开:
按照我上图的操作修改代码后,将原始的代码注释后保存,在运行代码就可以正常运行了。
2.2.3 测试
像上图这样就说明调通,代码已经在正常运行了。
三、总结
以上就是解决在使用YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample’ object has no attribute 'recompute_scale_factor’的快速解决方法。我个人更推荐第二种方法,快速高效,希望也能帮助到有这样问题的你!
感谢您阅读到最后!😊总结不易,多多支持呀🌹 点赞👍收藏⭐评论✍️,您的三连是我持续更新的动力💖
关注公众号「视觉研坊」,获取干货教程、实战案例、技术解答、行业资讯!