[发明专利]一种基于注意力机制的目标跟踪算法

权利要求书

1.一种基于注意力机制的目标跟踪算法，其特征在于，包括以下步骤：

步骤1：将两个连续的视频图像帧作为网络的输入，这两张图像通过相同的主干网络和BiFPN网络，得到五个不同空间分辨率的特征图；

步骤2：通过注意力机制，利用物体在当前帧中的位置来预测它在下一帧中的位置；

步骤3：在检测器失效或者发生遮挡的情况下，转而使用卡尔曼滤波器进行运动预测，并引入外观特征描述模块，通过重用步骤1中的特征，进行目标的重识别。

2.如权利要求1所述方法，其特征在于，步骤2中的注意力机制通过一种Non-LocalBlocks，实现一种注意力机制，使用两幅图像之间的注意力来进行更稳健的检测和运动预测。

3.如权利要求1所述方法，其特征在于，步骤3中的外观特征通过重用EfficientDet中的特征，使用Non-Local Blocks计算感兴趣区域的集中注意特征图。当需要在图像中提取表示一个对象的特征时，首先在五个特征映射和对象的边界框之间执行一个感兴趣区域的池化。然后利用整幅图像给与roi的注意力，计算roi。