【YOLOv5训练错误】权重文件出错?

AttributeError: Can‘t get attribute ‘SPPF‘ on <module ‘models.common‘ from ‘D:\\yolov5\\yolov5-5.0

AttributeError: Can‘t get attribute ‘SPPF‘ on <module ‘models.common‘ from

当出现上述错误时,我是这样考虑的:“yolov5 5.0”的模型中并没有使用SPPF,所以错误的出现应该是权值文件和模型的出了不匹配的问题。所以我单独下载了对应的权值文件,放入项目中,问题就解决了。

https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5m.pt

https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5l.pt

https://github.com/ultralytics/yolov5/releases/download/v5.0/yolov5s.pt

同理,需要v3.0或者v2.0版本的话,我们直接更改链接里的数字就行

写道这里突然想到一件事情,就是很多人在写论文时模型都参考江大白老师的,这没啥,可是人家那个版本是yolov5 2.0的,现在都6.0了,好多人画的都和他的一样???大人!时代变了啊!

Traceback (most recent call last):
  File "J:/yolov5-6.0xxs/train.py", line 620, in <module>

Traceback (most recent call last):
  File "J:/yolov5-6.0xxs/train.py", line 620, in <module>

这种情况说简单了就是没下载下来,搭个*就好了。


当然了,不使用初始训练权值也是可以的。。

 parser.add_argument('--weights', type=str, default=None, help='initial weights path')

上一篇:源码编译tensorflow


下一篇:编译tensorflow-serving错误no such package '@icu//'