基于改进YOLOv7-Tiny的变电设备红外图像识别

Infrared Image Recognition of Substation Equipment Based on Improved YOLOv7-Tiny Algorithm

  • 摘要: 针对复杂背景下变电设备红外图像目标识别精度不高、识别速度慢的问题,本文提出一种基于改进YOLOv7-Tiny的变电设备红外图像识别算法。首先,引入轻量级瓶颈结构GhostNetV2 BottleNeck替换部分CBS模块构建轻量级高效聚合网络L-ELAN(Lightweight-Efficient Layer Aggregation Network),同时在特征提取阶段嵌入CA(Coordinate Attention)注意力机制,在降低网络参数量的同时加强网络对目标关键特征的提取,提升检测精度;将网络坐标损失函数替换为SIoU Loss,以提升锚框定位精度和网络收敛速度;在变电设备红外数据集上进行测试,结果表明,改进后网络的精确率达到96.28%,检测速率达到26.42 frame/s,模型大小降低至7.82 M。与YOLOv7-Tiny原算法相比较,本文算法在提升识别精度的同时将检测速率提升21.69%,模型大小减少36.89%,可以满足变电站设备的精准实时识别要求,为后续的变电站设备故障诊断奠定基础。

     

    Abstract: To address the problem of low accuracy and slow recognition speed of infrared (IR) image target recognition of substation equipment in complex backgrounds, this study proposes an IR image recognition algorithm for substation equipment based on the improved YOLOv7-Tiny. First, the lightweight bottleneck structure GhostNetV2 bottleneck is introduced to replace a part of the CBS module and build a lightweight and efficient aggregation network known as a lightweight-efficient layer aggregation network. Simultaneously, a coordinate attention mechanism is embedded in the feature extraction stage to reduce the number of network parameters while strengthening the network's extraction of key features of the target and improving detection accuracy. The network coordinate loss function is replaced by SIoU_Loss to improve the anchor frame positioning accuracy and network convergence speed. The results show that the accuracy of the improved network is 96.28%, the detection rate is 26.42 frames/s, and the model size is reduced to 7.82 M. Compared with the original YOLOv7-Tiny algorithm, the detection rate is increased by 21.69%, the identification accuracy is improved, and the model size is reduced by 36.89%. These results meet the requirements of accurate real-time identification of substation equipment and lay a foundation for subsequent substation equipment fault diagnosis.

     

/

返回文章
返回