An innovative and improved method for infrared object detection is proposed, aiming to enhance the capabilities of the YOLOv8 model. This approach specifically addresses the model’s inherent limitations by integrating the Convolutional Block Attention Module (CBAM), which enhances feature discrimination through attention mechanisms. By focusing on the most relevant features in the infrared images, CBAM allows the model to make more accurate detections. Additionally, the method incorporates CLAHE (Contrast Limited Adaptive Histogram Equalization) as a preprocessing step. This technique significantly improves the overall interpretability of infrared images by enhancing contrast and making subtle features more visible. The combination of CBAM and CLAHE enables the model to better handle the challenges posed by infrared imaging, such as low contrast and noise. The mAP@0.5 obtained during this study for our adopted methodology is better than YOLO8, however the statistics stand at lower values of at 20.2 and 29.5% respectively.

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

Infrared Object Detection Method Based on YOLOv8 with CBAM Attention

  • Syed Muhammad Amir Raza

摘要

An innovative and improved method for infrared object detection is proposed, aiming to enhance the capabilities of the YOLOv8 model. This approach specifically addresses the model’s inherent limitations by integrating the Convolutional Block Attention Module (CBAM), which enhances feature discrimination through attention mechanisms. By focusing on the most relevant features in the infrared images, CBAM allows the model to make more accurate detections. Additionally, the method incorporates CLAHE (Contrast Limited Adaptive Histogram Equalization) as a preprocessing step. This technique significantly improves the overall interpretability of infrared images by enhancing contrast and making subtle features more visible. The combination of CBAM and CLAHE enables the model to better handle the challenges posed by infrared imaging, such as low contrast and noise. The mAP@0.5 obtained during this study for our adopted methodology is better than YOLO8, however the statistics stand at lower values of at 20.2 and 29.5% respectively.