一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统

    公开(公告)号:CN117710903B

    公开(公告)日:2024-05-03

    申请号:CN202410161198.5

    申请日:2024-02-05

    Abstract: 本发明公开了一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统,属于图像识别技术领域,所述方法包括以下步骤:(1)获取行人数据,并对数据进行预处理和存储;(2)对预处理后的数据进行逐帧提取,利用YOLOv5提取每帧图片中的person类图片并保存;(3)获取Market1501数据集随机划分测试集和训练集,并训练Resnet50模型,获取最终模型;(4)将已训练好的模型通过迁移学习方法,输入步骤(2)得到的图片,进行特征提取;(5)将提取后的图片合并为一个张量传递给ReID模型,并进行特征提取和归一化;(6)判定最小平均距离与距离阈值的大小;(7)保存最终视频;本发明通过YOLOv5对特定类的目标检测结合ReID对特定ID的检索,实现在协同检索追踪违章行人的功能。

    一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统

    公开(公告)号:CN117710903A

    公开(公告)日:2024-03-15

    申请号:CN202410161198.5

    申请日:2024-02-05

    Abstract: 本发明公开了一种基于ReID和Yolov5双模型下的可视化特定行人追踪方法及系统,属于图像识别技术领域,所述方法包括以下步骤:(1)获取行人数据,并对数据进行预处理和存储;(2)对预处理后的数据进行逐帧提取,利用YOLOv5提取每帧图片中的person类图片并保存;(3)获取Market1501数据集随机划分测试集和训练集,并训练Resnet50模型,获取最终模型;(4)将已训练好的模型通过迁移学习方法,输入步骤(2)得到的图片,进行特征提取;(5)将提取后的图片合并为一个张量传递给ReID模型,并进行特征提取和归一化;(6)判定最小平均距离与距离阈值的大小;(7)保存最终视频;本发明通过YOLOv5对特定类的目标检测结合ReID对特定ID的检索,实现在协同检索追踪违章行人的功能。

Patent Agency Ranking