yoloV5官方代码 运行报错|调参|训练自己的数据集
参数调整
- 修改data/coco.yaml(或者是自己的文件名) 例
- 修改model/yolov5s.yaml(或者是自己的文件名) 例
运行报错排查
dataset.py文件
问题描述:利用yolo v5代码训练自己的数据时出现AssertionError: train: No labels in autodl-tmp/PyQt5-YOLOv5-5.0/VOC2007/dataSet_path/train.cache. Can not train without labels. See https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data错误
在网上找了很多解决办法,有的说是路径问题,有的说将label改成labels,也有的说删掉train.cache和val.cache,都没能解决,废话不多说,直接上解决办法
**解决办法:**打开dataset.py文件,使用快捷键Ctrl+F使用搜索框搜索define label,将框中的地方,改为自己存放图片的文件夹名称,over✌
yolov5训练时,出现系数为nan和0的问题
gpu出现nan和0的问题。一般问题cuda问题和显卡的原因。
显卡为GTX 16XX系列的在cuda使用较新版本时会出现该问题。
解决方案为将cuda换为10.2的版本,链接如下,直接进行下载
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/torch_stable.html