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

2.报错models文件夹下common.py模块缺少SPPF类。
3.解决办法:
官方github,v6.0分支,去models-common.py下复制SPPF类:

到自己项目中common.py下,并
import warrings

二、
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.解决办法:
在运行配置中

取消勾选即可。

网友评论