计算机与现代化 ›› 2022, Vol. 0 ›› Issue (01): 70-76.

• 图像处理 • 上一篇    下一篇

融合注意力机制的轻量级红外高压套管识别算法

  

  1. (1.云南农业大学,云南昆明650201;2.华北电力大学,河北保定071000;3.河北工程大学,河北邯郸056000)
  • 出版日期:2022-01-24 发布日期:2022-01-24
  • 作者简介:国腾飞(1994—),男,河北邢台人,硕士研究生,研究方向:电力系统在线监测,E-mail: 1324703205@qq.com; 张则言(1996—),男,硕士研究生, 研究方向:电力系统在线监测,E-mail: 359888608@qq.com; 通信作者:付宏财(1966—),男,云南昆明人,副教授,硕士生导师,研究方向:新能源开发及电力系统分析,E-mail: fhc333@163.com; 王继选(1982—),男,副教授,硕士生导师,研究方向:能源利用及可再生能源应用,E-mail: wangjixuan113@163.com; 牛天宝(1995—),男,硕士研究生,研究方向:机器视觉及深度学习,E-mail: 1454334769@qq.com。
  • 基金资助:
    国家重点研发计划项目(2017YFD0300905-4); 河北省教育厅重点项目(ZD2020182); 河北省自然科学基金项目(E2017402084); 河北省科技厅科技攻关项目(17214509D); 河北省高等学校科学技术研究项目(ZD2021021)

Lightweight Infrared HV Bushing Identification Algorithm Based on Attention Mechanism

  1. (1. Yunnan Agricultural University, Kunming 650201, China; 2. North China Electric Power University, Baoding 071000, China;
    3. Hebei University of Engineering, Handan 056000, China)
  • Online:2022-01-24 Published:2022-01-24

摘要: 为了提高红外图像中变压器高压套管的识别准确率,同时能够满足移动端和其他低端设备对目标检测网络的需要,本文提出一种改进的轻量级红外高压套管识别算法,采用Tiny YOLOv3目标检测网络作为基础检测网络。首先,通过融合CBAM(Convolutional Block Attention Module)注意力机制,将通道注意力与空间注意力机制串联,增大目标检测网络感受野,同时减轻网络计算任务,提升网络性能;然后,分别使用GIoU loss和Focal loss替代原有的边界框损失和置信度损失,从而提高对红外图像中的高压套管识别率,减少漏检、误检情况发生。实验结果表明,改进的网络相比于原Tiny YOLOv3网络,mAP提升到96.28%,F1提升到96.25%,权重文件大小为33.9 MB,远小于YOLOv3训练网络,能够更好地适用于低端设备,为智能变电站的在线监测提供了有利条件。

关键词: 红外图像, 高压套管, Tiny YOLOv3, CBAM, GIoU loss, Focal loss

Abstract: In order to improve the identification accuracy of transformer HV bushing in infrared images and meet the needs of mobile terminal and other low-end devices for target detection network, this paper proposes an improved lightweight infrared HV bushing identification algorithm, using Tiny YOLOv3 target detection network as the basic detection network. First, through the fusion of the Convolutional Block Attention Module (CBAM) attention mechanism, the channel attention and the spatial attention mechanism are connected in series to increase the receptive field of the target detection network, while reducing network computing tasks and improving network performance. Then, GIoU loss and Focal loss are used to replace the original bounding box loss and confidence loss, thereby improving the recognition rate of the HV bushing in the infrared image and reducing the occurrence of missed and false detections. The experimental results show that compared with the original Tiny YOLOv3 network, the improved network increases mAP to 96.28%, increases F1 to 96.25%, and the weight size is 33.9 MB, less than that of YOLOv3 training network. It is better suitable for low-end equipment and provides favorable conditions for a smart substation online monitoring.

Key words: infrared images, high voltage bushing, Tiny YOLOv3, CBAM, GIoU loss, Focal loss