YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘的解决方法
在使用YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample’ object has no attribute 'recompute_scale_factor’的快速解决方法。
解决方法一:
有些博主说降低torchhe和torchvision版本,比如上图所示我的torch版本1.11.0 torchvision版本0.10.2,torch版本降低到版本1.9.1,torchvision版本降低到版本0.10.1。这是一种解决办法,但是要重新pytorch,我就嫌很麻烦,配置安装过程中可能又出现各种问题,所以我没有重新安装,采用了下面的这种方法,不用降低版本重载,就可以快速解决问题,见下。
解决方法二:
我们先看一下问题:
这个问题只要顺着路径找到文件upsampling.py并打开,修改里面的代码即可解决,见下:
按照我上图的操作修改代码后,将原始的代码注释后保存,在运行代码就可以正常运行了。
像上图这样就说明调通,代码已经在正常运行了。
以上就是解决在使用YOLOV5 训练好模型测试时出现问题:AttributeError: ‘Upsample’ object has no attribute 'recompute_scale_factor’的快速解决方法。我个人更推荐第二种方法,快速高效,希望也能帮助到有这样问题的你,谢谢!
2401_87012734: 怎么看对应版本啊
东^草: 你好,想用自己的数据集进行训练后部署需要在哪里进行改动呢
东^草: 你好,您是用自己的弧灰度图数据集训练之后部署了吗?成功了吗
视觉研坊: 不用谢
视觉研坊: 我源码包中提供的dataset.txt文件内,存放的是测试图片的文件名:space_shuttle_224.jpg。dataset这个变量是用来导入测试图片名称的,不能导入文件imagenet1000标签.txt。 源码包中的imagenet1000标签.txt文件是用来给你查看自己最后运行识别的类正不正确,imagenet1000标签.txt文件是该数据集官网提供的类别对应的索引和类名这些。