[发明专利]一种基于双流注意力机制的多模态监控视频目标跟踪方法

说明书

本发明属于目标跟踪技术领域，具体地说是涉及一种基于双流注意力机制的多模态监控视频目标跟踪方法。本发明以监控视频多模态数据的目标跟踪为背景，针对现有的多模态目标跟踪中对全局特征建模能力弱的缺点，提出了一种基于双流注意力机制的目标跟踪方法。该方法首先采用了双流注意力网络对目标的特征进行提取和融合，应用注意力机制对正负样本的关系进行建模，关注全局特征联系。此外，为了增强特征表达，对不同尺度的特征进行交互。最后利用回归分类子网络对目标的位置进行预测。这样既保证了局部细节，也关注了全局特征，得到了判别能力更强的特征，使回归分类子网络的预测结果更加精确。

技术领域

本发明属于目标跟踪技术领域，具体地说是涉及一种基于双流注意力机制的多模态监控视频目标跟踪方法。

背景技术

目标跟踪指的是在已知第一帧的目标的情况下，跟踪后续帧目标的位置。目标跟踪与传统的目标检测不同，不需要识别目标的种类，只需要识别目标在每一帧之中的位置。以往的目标跟踪多集中在可见光的监控视频数据中，但是在实际中，监控视频得到的可见光图像往往是清晰度不高的。此外，可见光视频由于其成像原因是光的反射，因此对光照十分敏感，光照过强的时候会产生曝光的问题，而光照过弱的时候则会看不清目标。因此，引入了红外图像对其做特征互补。红外图像的成像原理是利用不同物体的温度，因此对光照不敏感；但同时，红外图像本无法反映目标的细节，比如颜色、花纹等等。传统的多模态目标跟踪往往采用卷积神经网络对目标进行特征提取，他们使用分类网络单独处理学习每帧样本，而没有关注到正负样本之间的联系。这种方法的目标位置是通过目标分类的置信度得分得到的，可能使得目标的分类得分最高的地方并不是目标的位置。此外，不同深度的特征关注的特点也不相同。

发明内容

基于以上问题，本发明针对监控视频数据的目标跟踪，引入了可见光和红外图像的多模态融合互补，通过双流注意力机制来关注全局特征，并对不同尺度的特征进行融合，力求在保留局部细节的前提下，对目标和背景的联系进行关注，得到判别性更强的特征。

本发明以监控视频多模态数据的目标跟踪为背景，针对现有的多模态目标跟踪中对全局特征建模能力弱的缺点，提出了一种基于双流注意力机制的目标跟踪方法。该方法首先采用了双流注意力网络对目标的特征进行提取和融合，应用注意力机制对正负样本的关系进行建模，关注全局特征联系。此外，为了增强特征表达，对不同尺度的特征进行交互。最后利用回归分类子网络对目标的位置进行预测。这样既保证了局部细节，也关注了全局特征，得到了判别能力更强的特征，使回归分类子网络的预测结果更加精确。

本发明的技术方案是：

一种基于双流注意力机制的多模态监控视频目标跟踪方法，其特征在于，包括以下步骤：

S1、从监控视频数据中筛选出匹配的可见光-红外视频对，将视频对按照同一帧率保存为图像对获得输入样本；选取出图像对中代表第一帧的图像作为匹配图像，在匹配图像中将需要跟踪的目标用矩形框进行框选，后续帧的图像作为搜索图像；将输入样本记作i＝1,2,……N_j，N_j表示第j类样本的数量，表示可见光数据，表示红外数据；样本标签记作i＝1,2,……N_j，表示可见光数据标签，表示红外数据标签；在实际训练过程中，为了尽可能多的利用更多的数据，因此匹配分支随机选取视频序列M^train中的某一帧，记作(Z_v,Z_t)，搜索分支选择在匹配分支图像之后的某个图像对，记作(X_v,X_t)；

S2、构建神经网络，包括双流特征提取网络、多模态融合网络和跟踪网络；

所述双流特征提取网络包括两个结构相同分支，分别为匹配分支和搜索分支，匹配分支和搜索分支均分别对可见光图像和红外图像进行特征提取，具体为采用三个级联的transformer block进行特征提取；