yolo5是用于做目标检测的开源项目,检测又快又准,所以研究一下。
主要参考下面文章配置,在此多谢这篇文章的作者无私奉献,下面记录自己遇到的一些小问题。
https://blog.csdn.net/hh571050143/article/details/119968613
(29条消息) (包教会)配置Yolov5其实没那么复杂——Yolov5配置环境总结版_Mzgg的博客-CSDN博客
1 下载pytorch,我的主机没有GPU,所以我选择的CPU
官网链接: https://pytorch.org/get-started/locally/
2 测试Yolov5是否可以正常使用
在pycharm中选择我们配置好的环境,注意下面框中内容。
期间出现一个错误:AttributeError: 'Upsample' object has no attribute 'recompute_scale_factor';
百度一下,找到解决办法:
这个问题只要顺着路径找到文件upsampling.py并打开,修改里面的代码即可解决,见下:
按照我上图的操作修改代码后,将原始的代码注释后保存,在运行代码就可以正常运行了。
在教程中,进入我们的\yolov5-4.0\runs\detect\exp文件夹,没有图片,要自己加一行代码
我看我这版本中不是在53行,比对上面的图,我自己加了方框中两行。最终在run中出现了3个文件夹,在exp3中有正确的结果。