成功解决yolov5报错

1.环境配置问题

下载源码 用anaconda3新建一个环境

conda create -n yolov5 python==3.8

进入该新环境 activate yolov5

cd到源码文件夹，再进行pip install -r requirements.txt

2. 已装好cuda但运行python train.py --img 640 --batch 8 --epoch 300 --data data/VOC.yaml --cfg models/yolov5s.yaml --weights weights/yolov5s.pt --device ‘0,1‘

依旧报错

① 将--device'0,1'改成 --device 0 

② 修改VOC.yaml文件 

3.运行 python train.py --img 640 --batch 8 --epoch 300 --data data/VOC.yaml --cfg models/yolov5s.yaml --weights weights/yolov5s.pt --device 0报错

AssertionError: train: No labels in E:\yolov5-master\VOCdevkit\VOC2007\JPEGImages.cache. Can not train without labels. See https://github.com/ultralytics/yolov5/wiki/Train-Custom-Data

解决：打开E:\yolov5-master\utils datasets.py  ctrl+F 搜索define labels 将images改成自己修改的文件名 如这里使用voc数据集将 images改成JPEGimages

 

成功进入训练