yolov5 运行遇到的一些 bugs

一、

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