Infrared small target detection algorithm based on filter kernel combination optimization learning method

Abstract

Infrared small target detection in complex backgrounds remains a big challenge due to its small size and weak energy, which makes it easy to be submerged in complex backgrounds. Common filtering algorithms are difficult to apply to various complex scenes and generate many false alarms and poor robustness. Therefore, this paper proposes an infrared small target detection algorithm based on filter kernel combination optimization learning method. For the sample image block use typical filter kernels and weight coefficients to calculate the combination of filtering results, and construct the loss function of the results and the true value. The optimal weight coefficients are learned using an optimization method, which effectively achieves the fusion of multiple features, then we build a sample library containing comprehensive priori information. In the implementation, an online inference method based on similarity measure is proposed to match test images with sample library image blocks to obtain optimal combination weights. By utilizing the comprehensive priori information to enhance the adaptive ability of the algorithm for various complex backgrounds. Finally, get the position of the infrared small target. The experimental results show that our algorithm has excellent detection performance for infrared small targets with complex background and low signal to clutter ratio, and the AUC value is 0.9980, which is more robust than other comparative algorithms.