0
点赞
收藏
分享

微信扫一扫

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

爱薇Ivy趣闻 2022-01-31 阅读 85
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版本的话,我们直接更改链接里的数字就行

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')

举报

相关推荐

0 条评论