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
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容