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

Infrared Image Recognition of Substation Equipments Based on Improved YOLOv7

  • 摘要: 高效快速地识别变电站设备是变电站安全状态检测中至关重要的一个环节。针对变电站场景复杂且目标设备尺度不同的特点,在YOLOv7的基础上引入PSA模块,实现局部和全局通道之间的信息交互,提高模型对不同尺度设备的识别精度。再结合PConv和GSConv,建立轻量化网络,在确保模型精度的同时提升检测速度。使用Dyhead将3个感知嵌入一个目标检测头中,提升了目标的检测能力。构建变电站设备红外图像数据集,并进行训练、测试和验证,与原来的YOLOv7算法对比,准确率提升了3%,模型减小了10%,满足高效快速识别变电设备的要求,为后续变电设备故障诊断提供了基础。

     

    Abstract: Efficient and rapid identification of substation equipment is a crucial part in substation safety status detection. To better fit the complex backgrounds and different target devices of the substation, the PSA module is introduced on the basis of YOLOv7 to realize the information interaction between local and global channels and improve the recognition accuracy of the model for devices of different scales. Combined with PConv and GSConv, a lightweight network is established to accelerate the inspection progress while ensuring model accuracy. Using Dyhead to embed three perceptions into a target detection head, improving target detection capabilities. Compared with the original YOLOv7 algorithm, the accuracy is improved by 3% and the model is reduced by 10%, which meets the requirements of efficient and rapid identification of substation equipment and provides a basis for subsequent substation equipment fault diagnosis.

     

/

返回文章
返回