项目链接:https://github.com/ultralytics/yolov5/tree/v6.0
最近刚学完pytorch,想找几个项目跑一跑,今天上Github上找了个最火的YOLOv5的项目,下载了tags5版本,在运行其中的detect.py文件时又报错了(运行别人的项目时一定要注意先配置好环境):
说在models.py中找不到SPPF这个类,解决方法如下:
- 如果你用的是Tags5的话,就去Tags6里面的models/common.py里面去找到这个SPPF的类,把它拷过来到你这个Tags5的models/common.py里面,这样你的代码就也有这个类了,还要在Tags5的models/common.py里引入一个warnings包就可以了。