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
本次练习主要识别人,车辆,流动车摊