美文网首页
yolov5 运行遇到的一些 bugs

yolov5 运行遇到的一些 bugs

作者: 06am | 来源:发表于2021-12-01 19:57 被阅读0次

一、

1.初次运行v5.0版本的YOLOv5,自动下载权重yolov5s.pt

error
2.报错models文件夹下common.py模块缺少SPPF类。
3.解决办法:
官方github,v6.0分支,去models-common.py下复制SPPF类:
github-common.py
到自己项目中common.py下,并
import warrings
pycharm-common.py

二、

1.运行train.py报错:

RuntimeError: Given groups=1, weight of size [512, 1024, 1, 1], expected input[1, 512, 8, 8] to have 1024 channels, but got 512 channels instead

2.解决办法:
parser.add_argument('--cfg', type=str, default='', help='model.yaml path')
指定default=‘models/yolov5s.yaml’

三、

1.初次运行train.py,运行在python console中。
2.解决办法:
在运行配置中


configuration

取消勾选即可。


configuration

相关文章

网友评论

      本文标题:yolov5 运行遇到的一些 bugs

      本文链接:https://www.haomeiwen.com/subject/ccqexrtx.html