基于改进YOLOv8变电设备红外图像小目标检测

Electrical Equipment Infrared Image Small Target Detection Based on Improved YOLOv8

  • 摘要: 针对变电站电气设备里电力金具红外图像小目标呈现出特征不规则、纹理信息不足和识别精度不高等难题,本研究提出了一种基于YOLOv8模型的改进方法,融合多尺度扩张注意力机制(MSDA)和可变形卷积(DC),形成适应于红外图像小目标检测的动态上下文信息提取模块,从而有效获取目标全局特征与局部细节、动态适应红外小目标的形态变化。再通过嵌入协同注意力机制(CA),增强细节信息的提取能力和小目标检测能力,提升目标回归精度。同时,为解决网络卷积过程中细节特征丢失的问题,引入加权双向特征金字塔网络(BiFPN)模型,优化多尺度特征融合,防止目标特征信息被噪音遮盖。实验结果表明,经本文所提出方法改进后的YOLOv8模型在变电站红外图像小目标检测中,平均检测精度(mAP)提升了5.2%,召回率(Recall)提升了7.7%,精确率P提升了4.1%。

     

    Abstract: Aiming at the problem of irregular features,insufficient texture information,and low recognition accuracy in detecting small targets within infrared images of electrical equipment metal fittings in substations,an improved method based on the YOLOv8 model is proposed in this paper,which creates a Dynamic Contextual Information Extraction Module (DCIEM) suitable for small target detection in infrared images by integrating Multi-Scale Dilated Attention (MSDA) and Deformable Convolution (DC),so as to effectively obtain the global features and local details of the target and adapt to the morphological changes of small infrared targets.Then with embedded Coordinate Attention (CA) mechanism,the ability of detail information extraction and small target detection is enhanced to improve the accuracy of target regression.Meanwhile,in order to solve the problem of loss of detailed features in the process of network convolution,by introducing weighted bidirectional feature pyramid network (BiFPN) mode,a weighted bidirectional path aggregation network (Bi-PANet) is proposed to optimize multi-scale feature fusion and prevent the loss of original features.The experimental results show that the improved YOLOv8 model proposed in this paper,improves the mean detection accuracy (mAP) with increase of 5.2%,and the recall rate (Recall) with increase of 6.7%,and precision rate with increase of 4.1%.

     

/

返回文章
返回