yoloV5官方代码 运行报错|调参|训练自己的数据集

yoloV5官方代码 运行报错|调参|训练自己的数据集

• 参数调整
• 运行报错排查
• • dataset.py文件
  • yolov5训练时，出现系数为nan和0的问题

参数调整

• 修改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