Infrared Small Target Detection Algorithm Based on Super-Resolution Reconstruction and Improved YOLOv8n
Infrared small target detection primarily involves the recognition and localization of small targets within infrared images, holding great potential for various applications. This task faces several technical challenges due to the small size of targets in infrared images and the complexity of their backgrounds, including the low signal-to-noise ratio and limited pixels of the targets. We propose an improved YOLOv8n detection model based on sample super-resolution reconstruction to address these issues. First, the Spatially-Adaptive Feature Modulation (SAFMN) super-resolution method is employed to improve the resolution of the sample images and enhance the texture features of the targets. Subsequently, the processed images are cropped into smaller patches, and only patches containing targets are retained to strengthen the features of the infrared small targets and reduce the interference of the background. Finally, we integrate a parallelized patch-aware attention (PPA) module into the YOLOv8n network, which uses a multi-branch feature extraction strategy to capture features at different scales and levels of the targets. Experimental results show that, compared to the state-of-the-art detection methods, our method improves the mAP50 by 2.4% and Precision by 6.3% on the IRSTD1k dataset.
Authors
- Xiang Wu (ORCID: https://orcid.org/0000-0002-6305-1206)
- Huili Fan
- Yuming Shen
- Jian Yu
- Xingchen Zhang (ORCID: https://orcid.org/0000-0001-7264-4698)
Institutions
- Nanjing University of Science and Technology (CN)
- Wuhan Ship Development & Design Institute (CN)
Publication Details
- Published
- 2024-10-18
- DOI
- https://doi.org/10.1109/icocta64736.2024.00048
- Primary Topic
- Optical Systems and Laser Technology
- Type
- article
- Field-Weighted Citation Impact
- 0.00
Funders
- National Natural Science Foundation of China