使用YOLOv5训练自己的数据集

使用YOLOv5训练自己的数据

新东西出来，总是要试试，这几天使用了yolov5跑了几个模型，特来记录一下心得与坑。
作者给出的教程，可以先浏览一下: link.
源码下载链接: link.

一.环境部署

yolov5基于pytouch，在测试时发现与环境内的tensorflow会产生冲突，建议用ANACONDA新建pytouch的独立环境。
打开文件夹里的requirements.txt文件，用ANACONDA手动安装，或

pip install -U -r requirements.txt

二.数据集准备

yolov5使用的是yolo格式的标注文件，内容长这样，第一个数是标签的序号，后面四个是坐标。

标注软件依然是labelimg，在使用前将VOC格式转换为YOLO即可

如果有之前标注好的xml文件，可以通过脚本直接转成yolo所需的txt格式: link.
不过在转换完成后记得添加labels文件，标注文件根据序号从labels里面对应标签。

标注完后，建立如下目录，目录要求除了images/labels不同外，其他文件夹名