Infrared small target detection is a critical task in national defense, homeland security, aerospace, industrial monitoring, and environmental conservation, while the detection accuracy is generally insufficient due to the small size of the targets and the unclear features. Additionally, conventional detection algorithms often face challenges in terms of computational resources, making it difficult to achieve higher frame rates for infrared small target detection. To overcome these challenges, a new algorithm based on the latest YOLOv8 is introduced. In the feature extraction stage, the lightweight repeated C2f module from YOLOv8 is utilized, along with the incorporation of an attention mechanism. This approach reduces computational overhead while enhancing feature extraction capabilities. The detection results on the DIRST dataset demonstrate that the proposed method achieves an 8.4% increase in frame rate compared to the baseline network. Furthermore, it also exhibits improved performance in the confusion matrix analysis.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

YOLOv8 for Infrared Small Target Detection

  • Yilan Zhuo,
  • Wei Li,
  • Ju Huo,
  • Tao Chao

摘要

Infrared small target detection is a critical task in national defense, homeland security, aerospace, industrial monitoring, and environmental conservation, while the detection accuracy is generally insufficient due to the small size of the targets and the unclear features. Additionally, conventional detection algorithms often face challenges in terms of computational resources, making it difficult to achieve higher frame rates for infrared small target detection. To overcome these challenges, a new algorithm based on the latest YOLOv8 is introduced. In the feature extraction stage, the lightweight repeated C2f module from YOLOv8 is utilized, along with the incorporation of an attention mechanism. This approach reduces computational overhead while enhancing feature extraction capabilities. The detection results on the DIRST dataset demonstrate that the proposed method achieves an 8.4% increase in frame rate compared to the baseline network. Furthermore, it also exhibits improved performance in the confusion matrix analysis.