YOLOv5训练过程

该文介绍了如何结合YOLOv5和DeepSort进行车辆追踪项目的设置,包括从GitHub获取代码,准备数据集,修改yaml配置文件,调整train.py参数。YOLOv5用于对象检测,DeepSort则负责关联前后帧中的同一对象。
摘要由CSDN通过智能技术生成

课程链接【【YOLOv5+DeepSort+Pytorch】车辆追踪项目-使用DeepSort算法生成结果!-哔哩哔哩】 https://b23.tv/7B9iaoD

1、在GitHub下载yolov5-master代码

2、下载数据集,需要三种文件:

train、val,里面包含images,labels(txt格式)

 以及xxx(数据集名字).yaml,可以在数据集文件夹中,也可以在yolo代码的data文件下

 在xxx.yaml该文件中如下修改路径,其他的不管

 eg.

3、在train.py文件下调参

调整这三个参数

 

 在train.py文件下拉找到这三个参数:data中输入上述yaml文件的路径

总结:

1、数据集,包含train、val,里面都要分别包含图片(jpg)和标签(txt)

2、在data文件下新建一个yaml格式文件,内含train、val中图片的路径,还有nc(识别种类个数)、种类名称

3、在train.py修改配置

yolo和deepsort关系:yolo只能识别出每一帧中的行人,不负责判断前后两帧的行人的联系(是否是同一个人),deepsort承担此功能,联系起前后两帧行人的关系

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值