多源图像融合与改进YOLOv8的电气设备小目标故障识别

Multi-source Image Fusion and Improved Fault Recognition of Small Targets in Electrical Equipment by Improved YOLOv8

  • 摘要: 针对电气设备巡检中图像目标小、易受复杂背景及阴影遮挡干扰导致误检的问题,本文在对目标图像进行图像融合的基础之上提出了一种基于改进YOLOv8的电气设备故障检测算法。为减少在复杂背景下小目标故障的误检漏检,利用High-Order Recursive Network(HorNet)的递归门控卷积,设计用Convolutional Horblock with Customized Components(CHC)模块替代网络中原来的Cross Stage Partial Fusion with 2 convolutions(C2F)模块,加强了邻间特征的空间信息,并在骨干网络引入Efficient Multi-Scale Attention(EMA)多尺度注意力机制,捕捉更丰富的细节信息,提高了模型的特征提取能力。采用Wise-Intersection over Union(WIoU)损失函数替换原网络的损失函数,有效解决小目标漏检、误检、重叠目标被剔除等问题。实验结果表明,改进网络较原始YOLOv8n网络准确率提升19.8%、召回率提升10.8%、mAP50提升11.4%。本文提出的模型算法在小目标检测上具有更强的特征提取能力和更高的检测精度,适用于多源电气设备图像融合后的故障检测。

     

    Abstract: To solve the problem of the image target being small and susceptible to false detection caused by complex backgrounds and shadow occlusion interference in the inspection of electrical equipment, this study proposes an image fault detection algorithm for electrical equipment based on improved YOLOv8 based on image fusion of the target image. To reduce false detections and missed detections of small target faults in complex backgrounds, the recursive gated convolution of a High-Order Recursive Network (HorNet) was used to design a convolutional block with customized components (CHC), which replaces the original cross-stage partial fusion with two convolution (C2F) modules in the network, strengthens the spatial information of inter-neighborhood features, and introduces an Efficient Multi-Scale Attention (EMA) mechanism in the backbone network to capture richer detailed information and improve the feature extraction ability of the model. The Wise-Intersection over Union (WIoU) loss function was used to replace the original network loss function, effectively addressing the problems of missed detections, false detections, and overlapping target suppression. Experimental results show that, compared with the original YOLOv8n network, the improved network achieved a 19.8% increase in accuracy, a 10.8% increase in recall, and an 11.4% increase in mAP50. The proposed model demonstrates stronger feature extraction capability and higher detection accuracy for small targets and is suitable for fault detection in fused multi-source electrical equipment images.

     

/

返回文章
返回