Abstract:Addressing the formidable challenges in spatial infrared dim target detection, this paper introduces an advanced detection approach based on the refinement of the YOLOv8 algorithm. In contrast to the conventional YOLOv8, our method achieves remarkable improvements in detection accuracy through several novel strategies. Notably, by incorporating a deformable convolutional module into the YOLOv8 backbone network, our method effectively captures more intricate image features, laying a solid foundation for subsequent feature fusion and detection head predictions. Furthermore, a dedicated small target detection layer, built upon the original model, significantly enhances the model’s capability in recognizing infrared small targets, thereby boosting overall detection performance. Additionally, we utilize the WIoU-v3 as the localization regression loss function, effectively reducing sensitivity to positional errors and leveraging the advantages of multi-attention mechanisms. To enrich the quantity and quality of the spatial infrared dim target dataset, we employ image enhancement techniques to augment the original dataset. Extensive experiments demonstrate the exceptional performance of our method. Specifically, our approach achieves a precision of 95.6%, a recall rate of 94.7%, and a mean average precision (mAP) exceeding 97.4%, representing substantial improvements over the traditional YOLOv8 algorithm. Moreover, our detection speed reaches 59 frames/s, satisfying the requirements for real-time detection. This achievement not only validates the efficacy and superiority of our algorithm in spatial infrared dim target detection, but also offers novel insights and methodologies for research and applications in related fields, holding immense potential for future applications.

Research on Infrared Image Enhancement and Target Detection Algorithm Based on Deep Learning

Research on automatic target detection and recognition based on deep learning

Infrared target detection using deep learning algorithms

Infrared Weak and Small Target Detection Algorithm Based on Deep Learning

Infrared Small Target Detection Based on Local Contrast Vector and Signed Normalization

A Novel Infrared Small Target Detection Algorithm Based on Deep Learning

Infrared Small Target Detection Method Based on Frequency Domain Clutter Suppression and Spatial Feature Extraction

Review of Target Detection Technology based on Deep Learning

Research on Infrared Dim Target Detection Based on Improved YOLOv8

Infrared target recognition with deep learning algorithms

Infrared target detection and recognition in complex scene

Paying more attention to local contrast: improving infrared small target detection performance via prior knowledge

Infrared machine vision and infrared thermography with deep learning: A review

Deep-IRTarget: An Automatic Target Detector in Infrared Imagery Using Dual-Domain Feature Extraction and Allocation

Infrared Dim Small Target Detection Networks: A Review

EFLNet: Enhancing Feature Learning for Infrared Small Target Detection

Infrared Dim Target Detection Method Inspired by Human Vision System

Multiscale Progressive Fusion Filter Network for Infrared Small Target Detection

An Infrared Sequence Image Generating Method for Target Detection and Tracking

Image Enhancement-Based Detection with Small Infrared Targets

Infrared Small-Dim Target Detection with Transformer under Complex Backgrounds