基于改进YOLOv8的自适应特征增强红外目标检测算法

An Adaptive Feature Enhancement Algorithm for Infrared Object Detection Based on an Improved YOLOv8 Model

  • 摘要: 在红外目标检测中,由于目标与背景区分度差,往往导致检测精度低、漏检率高等问题。针对以上情况,提出了一种基于改进YOLOv8的红外目标检测方法。首先,设计了一种红外自适应特征增强模块(Infrared Adaptive Feature Enhancement Module, IAFEM),该模块主要对红外图像进行对比度增强和噪声抑制,进而输出增强后的特征,提升特征表达能力;其次,采用PIoU(Powerful-IoU)损失函数代替CIoU(Complete Intersection over Union)损失,使红外目标定位更精准,进一步提升检测精度。实验表明,改进后模型在FLIR、PDIWS(Person Detection in Intrusion Warning Systems)和HIT-UAV(Harbin Institute of Technology-Unmanned Aerial Vehicle)三种红外公开数据集上,mAP@0.5(Mean Average Precision)、mAP@0.5:0.95以及F1(F1-score)等指标相较于Base模型,分别提升3.3%、2.1%、3%,0.8%、1.9%、2%和2.7%、2.2%、1%,FPS(Frames Per Second)分别达到86.79、91.69和87.11。实验表明,改进模型有较好的实时性、泛化性和鲁棒性,并且有较好的检测效果。

     

    Abstract: In infrared target detection, poor contrast between the targets and backgrounds often leads to low detection accuracy and high false-negative rates. To address these issues, an infrared target detection method based on an improved YOLOv8 approach was proposed. First, an Infrared Adaptive Feature Enhancement Module (IAFEM) was designed. This module primarily enhanced contrast and suppresses noise in infrared images, thereby outputting enhanced features to improve the feature representation capabilities. Second, the CIoU loss function was replaced by the PIoU loss function, enabling precise infrared target localization and further enhancing the detection accuracy. Experiments demonstrated that the improved model achieved significant gains in precision, recall, mAP@0.5 (Mean Average Precision), mAP@0.5:0.95, and F1 (F1-score) over the base model by 3.3%, 2.1%, 3%, 0.8%, 1.9%, 2%, and 2.7%, 2.2%, 1%, respectively, with Frames Per Second (FPS) reaching 86.79, 91.69 and 87.11, respectively. The experiments demonstrated that the improved model exhibited good real-time performance, generalization, robustness, and superior detection performance.

     

/

返回文章
返回