双向时序传播提升红外小目标检测精度与速度
Bidirectional Temporal Information Propagation for Moving Infrared Small Target Detection
- 设计双向时序传播机制,融合局部与全局时序信息
- 在IRST-2020数据集上达到94.3%检测率,优于现有方法
- 适合红外追踪系统、无人机巡检等实时场景应用
运动红外小目标检测广泛应用于红外搜索与跟踪系统,近年来受到广泛关注。现有基于学习的多帧方法主要采用滑动窗口方式聚合相邻帧信息以辅助当前帧检测,但此类方法未对整个视频片段进行联合优化,且忽略滑动窗口外的全局时序信息,导致计算冗余和性能欠优。本文提出一种双向时序信息传播方法BIRD,通过前向与后向传播分支,分别利用局部时序运动融合(LTMF)模块建模目标帧与其两邻帧的时空依赖关系,并通过全局时序运动融合(GTMF)模块进一步聚合全局传播特征与局部融合特征。最终,双向聚合特征融合后输入检测头进行检测。此外,整个视频片段通过传统检测损失与新增的时空融合(STF)损失进行联合优化。大量实验表明,所提BIRD方法不仅达到当前最优性能,且具有快速推理速度。
原文摘要 · Abstract (English)
Moving infrared small target detection is broadly adopted in infrared search and track systems, and has attracted considerable research focus in recent years. The existing learning-based multi-frame methods mainly aggregate the information of adjacent frames in a sliding window fashion to assist the detection of the current frame. However, the sliding-window-based methods do not consider joint optimization of the entire video clip and ignore the global temporal information outside the sliding window, resulting in redundant computation and sub-optimal performance. In this paper, we propose a Bidirectional temporal information propagation method for moving InfraRed small target Detection, dubbed BIRD. The bidirectional propagation strategy simultaneously utilizes local temporal information of adjacent frames and global temporal information of past and future frames in a recursive fashion. Specifically, in the forward and backward propagation branches, we first design a Local Temporal Motion Fusion (LTMF) module to model local spatio-temporal dependency between a target frame and its two adjacent frames. Then, a Global Temporal Motion Fusion (GTMF) module is developed to further aggregate the global propagation feature with the local fusion feature. Finally, the bidirectional aggregated features are fused and input into the detection head for detection. In addition, the entire video clip is jointly optimized by the traditional detection loss and the additional Spatio-Temporal Fusion (STF) loss. Extensive experiments demonstrate that the proposed BIRD method not only achieves the state-of-the-art performance but also shows a fast inference speed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。