Unsupervised Image Sequence Registration and Enhancement for Infrared Small Target Detection
Runze Hou,Puti Yan,Xuguang Duan,Xin Wang
DOI: https://doi.org/10.1109/tgrs.2024.3392307
IF: 8.2
2024-05-03
IEEE Transactions on Geoscience and Remote Sensing
Abstract:In the burgeoning realm of deep learning (DL) and with the introduction of the infrared target detection dataset, infrared small target detection has increasingly garnered attention. Within this domain, multiframe infrared small target detection stands as a pivotal and challenging subtask. Notably, some recent methods have exhibited commendable performance in multiframe infrared scenes. However, these methods were tethered to time-consuming background alignment preprocessing, which impedes their real-world application of multiframe infrared target detection systems. In this article, an unsupervised end-to-end framework tailored for infrared image sequence registration was proposed. This framework diverges from traditional registration methods by incorporating a novel basket-based hierarchical temporal consistency loss. The proposed loss function achieves intrabasket consistency and interbasket diversity (IBD), effectively mitigating issues related to inconsistency. In addition, the framework includes the input thresholding mask and output transformation mask. These components are crucial for guiding the network's training process and correcting misalignments. Moreover, the introduction of a dual-level residual enhancer is proposed to enhance the quality of registered images, thereby improving overall performance. Extensive experimental results have demonstrated the superiority of the proposed method over baseline methods. The proposed method achieved a significant improvement in the metric on a public dataset, reaching 0.8882, and an inference speed of 23.34 FPS. This represents an improvement of 0.0190 in performance and a sixfold increase in speed compared to the state-of-the-art (SOTA) method in multiframe infrared small target detection.
engineering, electrical & electronic,imaging science & photographic technology,remote sensing,geochemistry & geophysics