公开(公告)号：CN119090917A

公开(公告)日：2024-12-06

申请号：CN202411209061.9

申请日：2024-08-30

Applicant: 浙江大学

Inventor： 戴宇晗 ,  王闻箫 ,  林彬彬 ,  蔡登

IPC: G06T7/246 ,  G06V10/25 ,  G06V10/40 ,  G06V10/75 ,  G06V10/764 ,  G06V10/82 ,  G06V20/40 ,  G06N3/042 ,  G06N3/0442 ,  G06N3/045 ,  G06N3/0464 ,  G06N3/08

Abstract: 本发明公开了一种基于时间记忆和GNN的多目标跟踪方法，包括训练LSTM、GhostNetV3、GCN模型，使用预训练的YOLOX对视频的每一帧进行目标检测得到检测框；将每一个跟踪到的方框送入对应的LSTM网路中得到预测框，拿YOLOX得到的检测框和LSTM得到的预测框做匹配，对当前帧的目标检测结果进行分类，分别做不同的处理；本发明使用YOLOX作为检测器、使用LSTM来进行预测、使用GCN高效地处理多节点之间复杂的关系，使用GhostNetV3保证特征提取效果的同时降低计算复杂度，使本发明方法在保证运行精度的前提下，能更快提取出特征，仍然保持着较高的帧率和舒适的视觉效果。