海思hi3516dv300芯片AI图像模块学习（三）YOLO训练自己的数据集

参考 ：https://gitee.com/losschen/yolov5-pytorch/blob/main/README.md 

一、制作数据集文件

根目录下创建数据集文件夹，类别文档

图片和标注扔到Annotations和JPEGimages里面

把MAIN文件夹从voc数据集里复制过来

打开voc_Annotations.py

改路径

由于文件夹取名GangMao2022，year2007改2022

每一个%s前的VOC都改成GangMao

改

保存运行，报错

改成正确的标签名字

成功，新的文件信息生成了。

二、训练yolo模型

参考：

运行报错

 在pytorch环境里面下载一下，pip install tensorboard 

报错

FileNotFoundError: [Errno 2] No such file or directory: '2007_train.txt'

注意路径不是数据集里那个，而是主目录下新生成的2022_train.txt

里面的样子

优化方法名字打错了

adam

ok，成功运行，推荐577，但是先跑100世代试一试。

昨晚跑了7个小时的结果，使用测试集的效果还可以但是换一般的图片识别率就不行了，并且首片的识别率还是太低。