面向天基红外预警的高动态弱小目标LSTM检测方法研究

Research on LSTM method of high dynamic dim and small targets detection for space-based infrared early warning

  • 摘要: 在天基红外预警任务中,高动态弱小目标具有成像尺寸小、运动规律未知的特点。现有红外弱小目标探测任务主要关注匀速直线运动目标检测问题,对高动态目标的有效检测算法尚需进一步开发。针对天基红外预警任务中高动态非线性运动目标检测问题,提出了一种基于LSTM的红外弱小目标检测算法。首先设计了提取可疑目标位置信息的预处理方法,解决了LSTM网络结构与序列图像不匹配的问题;然后,针对传统算法难以检测非线性运动轨迹的问题,利用LSTM提取目标运动特征,实现序列图像中高动态目标的检测。通过与序列假设检验等算法的对比,在自研的红外序列图像数据集上验证了所提出的算法能够实现不低于0.9347的精确率与不低于0.8633的召回率。

     

    Abstract:
      Objective  Infrared target detection technology has the advantages of large observation range, short scanning period and strong anti-interference ability, which is of great significance in space-based early warning missions. The infrared targets in space-based early warning missions have the characteristics of small size and weak signal, and it is difficult to use their texture and shape features for detection. In scenes with strong noise and clutter interference, it is necessary to use target motion features for detection. The traditional target detection algorithm generally assumes that targets move with constant velocity, which is not capable of detecting high dynamic nonlinear moving targets with unknown motion rules, and effective detection algorithms for nonlinear moving targets still need further development. To solve the problem of high dynamic and nonlinear moving target detection in space-based early warning missions, an infrared dim and small target detection algorithm based on LSTM is proposed.
      Methods  Firstly, an adaptive preprocessing block is proposed, which can extract the location information of suspicious targets and solves the problem of mismatch between LSTM network structure and sequence images. At the same time, some spatial characteristics of target signals are discarded in order to reduce the amount of computation. Then, due to the problem that traditional algorithms are unable to detect nonlinear motion trajectory, a target trajectory detection method based on LSTM is designed to realize the detection of high dynamic nonlinear moving targets in sequence images. Finally, a post-processing algorithm, which selects target points from the pre-processing results based on the LSTM estimation results is designed to improve the target positioning accuracy.
      Results and Discussions   Aiming at the targets with different motion laws, two experiments are designed and carried out based on the image sequence with an average signal-to-noise ratio of 4.07. In the first experiment, the target performs simple nonlinear motion with a sinusoidal law. This experiment proves that the proposed algorithm can correctly estimate the existence and position coordinates of the target with a precision rate of 0.934 7 and a recall rate of 0.885 1 (Fig.6-7). In the second experiment, the target performs complex nonlinear motion described by the superposition of multiple sine functions, the proposed algorithm achieves a precision rate of 0.936 2 and a recall rate of 0.863 3 at a detection speed of 0.008 326 s/frame and a peak memory usage of 1 214.13 MiB (Tab.2). Compared with four algorithms including Sequential Hypothesis Testing, the proposed algorithm is proved to have better performance in terms of precision rate, recall rate and detection speed. Through some additional experiments, it is proved that the algorithm is applicable to the detection task in the scenes where the average speed of target is between 1-6 pixels/frame and the average SCR and SNR of sequence images are both higher than 2.
      Conclusions  Experiments results demonstrate that the proposed algorithm has the ability to detect dim and small targets with high dynamic nonlinear motion from noise and clutter. The proposed algorithm makes use of the spatial and temporal features of the target in the sequence images, and is able to detect targets in low signal-to-noise ratio scenes. And at the same time, with the help of LSTM's ability to extract temporal features, the proposed algorithm is capable in the task of nonlinear moving targets detection. In addition, as the lightweight structure of the LSTM is maintained, the proposed algorithm shows high real-time performance.

     

/

返回文章
返回