Infrared Weak and Small Target Detection Algorithm Based on Deep Learning

Abstract

Abstract In the infrared imaging scene where the target is at a long distance and the background is cluttered, due to the interference of noise and background texture information, the infrared image is prone to problems such as low contrast between the target and the background, and feature confusion, which makes it difficult to accurately extract and detect the target. To solve this problem, firstly, the infrared image is enhanced by combining DDE and MSR algorithm to improve the contrast and detail visibility of the image. For the RT-DETR network structure, the EMA attention mechanism is introduced into the backbone to enhance the feature extraction ability of the model by extracting context information. The CAMixing convolutional attention module is introduced into CCFM, and the multi-scale convolutional self-attention mechanism is introduced to focus on local information and enhance the detection ability of small targets. The filtering rules of the prediction box are improved, combined with Shape-IoU, and the convergence speed of the loss function in the detection and the detection accuracy of small targets are improved by paying attention to the influence of the intrinsic properties of the bounding box itself on the regression. In the experiment, the infrared weak target image dataset of the National University of Defense Technology was selected, labeled and trained. Experimental results show that compared with the original DETR algorithm, the average precision of the improved algorithm (mAP) is increased by 3.2%, and it can effectively detect infrared weak and small targets in different complex backgrounds, which reflects good robustness and adaptability, and can be effectively applied to infrared weak and small target detection in complex backgrounds.