yolov5 运行遇到的一些 bugs
2021-12-01 本文已影响0人
06am
一、
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
