Infrared small target detection based on multiscale local contrast learning networks

Abstract

Recently, model-driven deep networks have achieved excellent detection performance on infrared small targets in cluttered environments. However, its detection performance is sensitive to the hyperparameters in the embedded model-driven module. Therefore, we propose a novel multiscale local contrast learning network (MLCL-Net), which is an end-to-end fully convolutional infrared small target detection network. By constructing a local contrast learning (LCL) structure, it can learn to generate local contrast feature maps during training. Considering the difference in target size, we further build a multiscale local contrast learning (MLCL) module based on LCL. By extracting and fusing local contrast information of different scales from feature maps of the same level, the feature information of targets is fully excavated. At the same time, due to the small size of the target, a slight pixel shift will cause a severe loss of accuracy. We propose a bilinear feature pyramid network (BFPN) based on the feature pyramid network (FPN). Compared to state-of-the-art methods, the proposed MLCL-Net achieves superior performance with an intersection-over-union (IoU) of 0.772 and normalized IoU (nIoU) of 0.755 on the public SIRST dataset.