项目中提供了行人ReID模型权重,放在yolov5-deepsort\deep_sort\deep_sort\deep\checkpoint
下名为ckpt.t7
的权重文件,是基于deepsort在行人ReID数据集训练出来,用于提取行人的外观特征
。
如果你自己有更好的ReID数据集的话,是可以重新训练的,本文将介绍在Marker-1501
数据集下进行行人ReID训练
Market 1501 下载链接:/s/1dlG2CX6ZQKCW8cp9uwptEw?pwd=rwzn
提取码:rwzn
1. 行人ReID训练
1.1 Market 1501数据集
Market-1501 数据集是在清华大学
校园中采集于构建并公开。它包括由6个摄像头
(其中5个高清摄像头和1个低分辨率摄像头)拍摄到的1501
个行人、32668
个检测到的行人矩形框。每个行人
至少有2个
摄像头捕捉到,并且在一个摄像头中可能具有多张图像。训练集有751
人,包含12936张图像,平均每个人有17.2张训练数据;测试集有750
人,包含19732张图像,平均每个人有26.3张测试数据。