基于多相似度矩阵融合的多目标跟踪方法和装置

一种基于多相似度矩阵融合的多目标跟踪方法和装置，其方法包括：将输入图像输入目标检测模型，得到图像中每个目标的检测框，通过卡尔曼滤波器对上一帧目标的运动轨迹进行预测，得到预测框；将检测框和预测框输入到外观模型WRN和SIFT特征提取器，得到对应的SIFT特征和WRN特征；计算检测框和预测框目标之间的马氏距离和余弦距离，构建出对应的WRN特征相似度矩阵和SIFT特征相似度矩阵，按照一定权重进行融合得到最终的代价矩阵；通过匈牙利算法对目标进行匹配，得到第一次匹配结果，将第一次匹配结果中未匹配的目标和未匹配的检测再进行一次IOU匹配，得到目标的轨迹；通过与外观无关的连接模型和高斯平滑插值法，得到目标最终的轨迹。本发明在目标尺寸小、遮挡、模糊等复杂场景下展现了良好的性能。