基于跨相机的多目标跟踪模型的训练方法及装置

    公开(公告)号:CN115393384A

    公开(公告)日:2022-11-25

    申请号:CN202211116422.6

    申请日:2022-09-14

    Applicant: 清华大学

    Abstract: 本申请提供了基于跨相机的多目标跟踪模型的训练方法及装置,包括:获取多个训练样本组合,所述训练样本组合包括一个场景的多个相机图像样本,所述相机图像样本上标注多个目标的真实框;利用多目标跟踪模型对每个训练样本组合进行处理,得到多个相机图像的预测结果,每个相机图像的预测结果包括多个目标的预测框;基于多个相机图像的预测结果和多个相机图像样本,分别计算吸引项损失函数值、第一排斥项损失函数值以及第二排斥项损失函数值;将三者的加权和作为总损失函数值;基于总损失函数值,更新多目标跟踪模型的模型参数。本申请可以有效解决目标遮挡情况下的无法实现多目标跟踪的技术问题。

Patent Agency Ranking