基于轻量化网络与注意力融合的变电设备红外缺陷识别

Infrared Defect Image Recognition Method for Substation Equipment Based on Attention Fusion and Lightweight Network

  • 摘要: 针对复杂背景变电设备红外图像下的缺陷处信息模糊、对比度低导致识别精度低,部署移动端识别速度慢的问题,本文提出了一种YOLOv8-GCAFM(YOLOv8 based on ghost and attention mechanism fusion)变电设备红外缺陷识别算法。首先在骨干网络中引入轻量级GhostNetV2 BottleNeck替换部分CBS模块,减少网络参数数量,增强模型的表达能力,提升检测速度;其次提出了一种注意力融合模块v3_CAFM,添加到颈部网络,使在全局和局部捕获更广泛的特征信息,提高网络精度;最后采用Wise Intersection over Union(WIoU)损失函数优化梯度增益分配策略,增强检测器对多尺度目标变化的适应性。在包含多种变电设备红外缺陷图像的自建数据集上进行训练。实验结果表明,改进后算法与基线算法相比mAP@0.5、mAP@0.5:0.95、P分别提升了3.2%、8.1%、3.8%,模型大小为6.1MB,性能超过YOLOv8。YOLOv8- GCAFM模型能够满足设备缺陷的实时识别要求,为后续变电设备的故障诊断及在移动端等资源受限设备上应用提供参考。

     

    Abstract: Aiming at overcoming the problems of fuzzy information and low contrast at the defect in infrared images of substation equipment with complex backgrounds, which lead to low recognition accuracy and slow identification speed in mobile terminal deployment, this study developed an infrared defect identification algorithm (YOLOV8-GCAFM: YOLOv8 based on Ghostv2 and attention mechanism fusion). First, a lightweight GhostNetV2 BottleNeck was introduced in the backbone network to replace some CBS modules, which reduced the number of network parameters, enhanced the model expression ability, and increased the detection speed. Second, an attention fusion module (v3_CAFM) was developed, which was added to the neck network to capture more extensive feature information globally and locally, and improve network accuracy. Finally, the wise intersection over union loss function was used to optimize the gradient gain allocation strategy to enhance the adaptability of the detector to changes in multiscale targets. A test was conducted on a self-built dataset containing infrared defect images of various pieces of transformer equipment. The experimental results showed that, compared with the baseline algorithm, mAP@0.5, mAP@0.5:0.95, and P/% increased by 3.2%, 8.1%, and 3.8%, respectively. The model size was 6.1 MB, and the performance of the improved algorithm exceeded that of YOLOv8. The YOLOv8-GCAFM model could satisfy the requirements of the real-time identification of equipment defects and provide a reference for the subsequent fault diagnosis of substation equipment and applications on resource-limited devices such as mobile terminals.

     

/

返回文章
返回