一种基于目标类别改进的多目标跟踪方法

专利主权项内容

1.一种基于目标类别改进的多目标跟踪方法，其特征在于，包括以下步骤：S100，获取视频vid中第i帧图像P对应的预测目标框集合A，A＝{a, a, …, a, …, a}，a为P对应的第n个预测目标框，n的取值范围为1到N，N为P对应的预测目标框的数量；i12nNniiS200，获取vid中第i+1帧图像P对应的检测目标框集合B，B＝{b, b, …, b, …, b}，b为P对应的第m个检测目标框，m的取值范围为1到M，M为P对应的检测目标框的数量；i+112mMmi+1i+1S300，获取IOU矩阵T，T为M×N的矩阵，T的第m行第n列的元素t为b与a的交并比；m, nmnS400，获取目标类别矩阵V，V为M×N的矩阵，V的第m行第n列的元素v为b与a的类别权值；当b对应的目标类别与a对应的目标类别为相同类别或易混淆的类别时，v＝1；当b对应的目标类别与a对应的目标类别为不同且不易混淆的类别时，v＝0；m, nmnmnm, nmnm, nS500，利用V对T进行更新，得到更新后的目标IOU矩阵T’，T’为M×N的矩阵，T’的第m行第n列的元素为t’，t’＝t×v；m, nm, nm, nm, nS600，根据T’获取损失矩阵L，L为M×N的矩阵，L的第m行第n列的元素为l，l＝1-t’；m, nm, nm, nS700，根据L获取A和B中目标框的匹配结果。