首页 | 本学科首页   官方微博 | 高级检索  
     

融合注意力机制的轻量级红外高压套管识别算法
引用本文:国腾飞,张则言,付宏财,王继选,牛天宝. 融合注意力机制的轻量级红外高压套管识别算法[J]. 计算机与现代化, 2022, 0(1): 70-76. DOI: 10.3969/j.issn.1006-2475.2022.01.010
作者姓名:国腾飞  张则言  付宏财  王继选  牛天宝
作者单位:云南农业大学,云南 昆明 650201,华北电力大学,河北 保定 071000,河北工程大学,河北 邯郸 056000
基金项目:国家重点研发计划项目(2017YFD0300905-4);河北省教育厅重点项目(ZD2020182);河北省自然科学基金项目(E2017402084);河北省科技厅科技攻关项目(17214509D);河北省高等学校科学技术研究项目(ZD2021021)。
摘    要:为了提高红外图像中变压器高压套管的识别准确率,同时能够满足移动端和其他低端设备对目标检测网络的需要,本文提出一种改进的轻量级红外高压套管识别算法,采用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  
收稿时间:2022-01-24

Lightweight Infrared HV Bushing Identification Algorithm Based on Attention Mechanism
GUO Teng-fei,ZHANG Ze-yan,FU Hong-cai,WANG Ji-xuan,NIU Tian-bao. Lightweight Infrared HV Bushing Identification Algorithm Based on Attention Mechanism[J]. Computer and Modernization, 2022, 0(1): 70-76. DOI: 10.3969/j.issn.1006-2475.2022.01.010
Authors:GUO Teng-fei  ZHANG Ze-yan  FU Hong-cai  WANG Ji-xuan  NIU Tian-bao
Affiliation:(Yunnan Agricultural University,Kunming 650201,China;North China Electric Power University,Baoding 071000,China;Hebei University of Engineering,Handan 056000,China)
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.
Keywords:infrared images  high voltage bushing  Tiny YOLOv3  CBAM  GIoU loss  Focal loss
本文献已被 维普 万方数据 等数据库收录!
点击此处可从《计算机与现代化》浏览原始摘要信息
点击此处可从《计算机与现代化》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号