课程链接【【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承担此功能,联系起前后两帧行人的关系