基于transformer的时空信息融合的红外目标跟踪方法

作者:齐美彬; 汪沁昕; 庄硕; 张可; 李坤袁; 刘一敏; 杨艳芳
来源:2023-04-11, 中国, ZL202310406030.1.

摘要

本发明公开了一种基于transformer的时空信息融合的红外目标跟踪方法,包括:第一步,对红外图像进行预处理;第二步,构建红外目标跟踪网络,包括:红外图像特征提取子网络、红外图像特征融合子网络、角点预测头子网络、显著点聚焦子网络、IOU-Aware目标状态评估头子网络;第三步,构建红外目标跟踪网络的损失函数;第四步,采用两阶段训练法优化红外目标跟踪网络。本发明通过设计多个组件来实现红外目标跟踪过程中时空信息的融合,旨在提高不同跟踪场景下红外目标跟踪方法的准确性和鲁棒性。