Yolo V5 训练

1、下载YOLO V5源代码

git clone https://github.com/ultralytics/yolov5

2、使用yolv v5进行图片检测

--save-txt：会将检测结果以txt的形式保存

内容依次为：<object-class> <x_center> <y_center> <width> <height>

如：56 0.758585 0.868902 0.0762363 0.140244

python detect.py --source images/lbjy_1.jpg --weights  weight/yolov5s.pt --save-txt

3、生成自己的训练数据，进行模型训练

3.1 标记数据，我用的labelImg进行图片标记

3.2 修改目录结构

默认在yolov5同一目录下新建一个datasets目录，用于存放训练数据（参考Train-Custom-Data）

3.3 编辑配置文件 data/wg.yaml

本次练习主要识别人，车辆，流动车摊