一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统
申请人信息
- 申请人:南京信息工程大学
- 申请人地址:210044 江苏省南京市江北新区宁六路219号
- 发明人: 南京信息工程大学
专利详细信息
| 项目 | 内容 |
|---|---|
| 专利名称 | 一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统 |
| 专利类型 | 发明申请 |
| 申请号 | CN202410161198.5 |
| 申请日 | 2024/2/5 |
| 公告号 | CN117710903A |
| 公开日 | 2024/3/15 |
| IPC主分类号 | G06V20/52 |
| 权利人 | 南京信息工程大学 |
| 发明人 | 庄建军; 万超; 庄宇辰; 万理; 陈永强; 王楠; 庞明义 |
| 地址 | 江苏省南京市江北新区宁六路219号 |
摘要文本
本发明公开了一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统，属于图像识别技术领域，所述方法包括以下步骤：(1)获取行人数据，并对数据进行预处理和存储；(2)对预处理后的数据进行逐帧提取，利用YOLOv5提取每帧图片中的person类图片并保存；(3)获取Market1501数据集随机划分测试集和训练集，并训练Resnet50模型，获取最终模型；(4)将已训练好的模型通过迁移学习方法，输入步骤(2)得到的图片，进行特征提取；(5)将提取后的图片合并为一个张量传递给ReID模型，并进行特征提取和归一化；(6)判定最小平均距离与距离阈值的大小；(7)保存最终视频；本发明通过YOLOv5对特定类的目标检测结合ReID对特定ID的检索，实现在协同检索追踪违章行人的功能。
专利主权项内容
1.一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法，其特征在于，包括以下步骤：(1)利用若干台1280*1080高清摄像机和SD摄像机获取行人数据，并对数据进行预处理并进行存储；(2)对预处理后的数据进行逐帧提取，利用YOLOv5提取每帧图片中的person类图片并通过提取person类图片坐标进行图片保存；(3)获取Market1501数据集随机划分测试集和训练集，并训练Resnet50模型，获取最终模型；(4)将已训练好的模型通过迁移学习方法，输入步骤(2)得到的图片，进行特征提取；(5)将提取后的图片合并为一个张量传递给训练好的ReID模型，并进行特征提取和归一化，通过计算对query图片与每个保存图片之间的欧几里得距离的均值，找到最小平均距离的索引；(6)通过不同场景设置相应距离阈值，判定最小平均距离与距离阈值的大小，当最小平均距离比距离阈值小时，将最终的目标边界框输出，包括边界框的位置坐标、置信度和类别信息；(7)保存最终视频。