基于改进的YOLOv8变电设备红外图像检测

Infrared Image Detection of Substation Equipment Based on Improved YOLOv8

  • 摘要: 针对变电站电气设备红外图像远距离、小目标的检测特点,本文提出了一种改进的YOLOv8模型检测方法。改进后的YOLOv8模型引入了CA注意力机制模块,增强了网络模型的特征提取和特征融合能力,将YOLOv8原模型的损失函数CIoU改进为SIoU,以降低模型的误判率,并添加一个针对远距离、小目标的检测头YOLO-Head。通过对YOLOv8模型在以上3个方面的改进,变电设备红外图像目标检测的平均检测精度mAP从89.28%提高到93.57%,其P-R(精准度-召回率)综合指标提高了3.2%。

     

    Abstract: Infrared detection of substation electrical equipment is challenging owing to remote imaging conditions and small-target sizes. To address these challenges, this study proposes an improved YOLOv8-based detection method. In the improved YOLOv8 model, the Coordinate Attention (CA) mechanism module was introduced to strengthen feature extraction and feature fusion capabilities of the network model. The original model loss function, CIoU, was optimized to SIoU to reduce misjudgments of the model, and the detection head, YOLO-Head, was supplemented to improve detection of remote and small targets. With these three improvements, the average detection accuracy, expressed as mean Average Precision (mAP), on infrared images of substation electrical equipment increased from 89.28% to 93.57%, and the comprehensive index of precision-recall rate (P-R) also increased by 3.2%.

     

/

返回文章
返回