Yolo V5 训练

本文介绍了如何使用YOLOv5进行目标检测,从下载源代码到利用labelImg标记数据,修改目录结构,编辑配置文件,并进行模型训练。在训练过程中,遇到大量样本导致的内存问题,提出了通过设置Docker共享内存解决方法。
摘要由CSDN通过智能技术生成

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

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


                
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值