计算机科学 ›› 2022, Vol. 49 ›› Issue (6A): 412-417.doi: 10.11896/jsjkx.210600089

• 图像处理&多媒体技术 • 上一篇    下一篇

基于GDIoU损失函数的YOLOv4绝缘子高效定位算法

马宾, 付永康, 王春鹏, 李健, 王玉立   

  1. 齐鲁工业大学(山东省科学院) 济南 253000
    山东省计算机网络重点实验室 济南 253000
  • 出版日期:2022-06-10 发布日期:2022-06-08
  • 通讯作者: 马宾(mab@qlu.edu.cn)
  • 基金资助:
    国家自然科学基金(61802212,61872203);山东省自然科学基金(ZR2019BF017,ZR2020MF054);山东省高校科研计划项目(J18KA331);山东省重大科技创新工程项目(2019JZZY010127,2019JZZY010132,2019JZZY010201);山东省高等学校青创人才引育计划(SD2019-161);济南市“高校20条”引进创新团队(2019GXRC031)

High Performance Insulators Location Scheme Based on YOLOv4 with GDIoU Loss Function

MA Bin, FU Yong-kang, WANG Chun-peng, LI Jian, WANG Yu-li   

  1. Qilu University of Technology(Shandong Academy of Sciences),Jinan 253000,China
    Shandong Provincial Key Laboratory of Computer Networks,Jinan 253000,China
  • Online:2022-06-10 Published:2022-06-08
  • About author:MA Bin,born in 1973,Ph.D,professor,is a member of China Computer Federation.His main research interests include information hiding and multimedia security and digital image processing.
  • Supported by:
    National Natural Science Foundation of China(61802212,61872203),Shandong Provincial Natural Science Foundation(ZR2019BF017,ZR2020MF054),Major Scientific and Technological Innovation Projects of Shandong Province(2019JZZY010127,2019JZZY010132,2019JZZY010201),Plan of Youth Innovation Team Development of Colleges and Universities in Shandong Province(SD2019-161) and Jinan City “20 universities” Funding Projects Introducing Innovation Team Program(2019GXRC031).

摘要: 绝缘子检测是保障输电系统安全稳定的重要措施,绝缘子定位是进行检测的前提。针对目前电力巡检中绝缘子定位速度慢、精度低的问题,提出了一种基于GDIoU(Gaussian Distance Intersection over Union)损失函数的YOLOv4深度学习框架。该方案通过设计GDIoU损失函数来提高YOLOv4的定位精度和收敛速度,利用二维高斯模型提高了网络的收敛能力,增强了YOLOv4的性能,进而提高了绝缘子的定位精度与速度。同时提出绝缘子自适应旋转矫正算法,通过对单个绝缘子图像进行旋转矫正,提升了在不同空间状态下的绝缘子识别精度。实验结果表明,与朴素YOLOv4相比,所提算法的定位精度提高了7.37%。在同水平的精度下,基于GDIoU的YOLOv4绝缘子定位方法比其他绝缘子定位算法速度快了3倍以上。所提方法在精度与速度上做了较好的平衡,其性能完全满足电力巡检中绝缘子的在线定位要求。

关键词: GDIoU Loss, 电力巡检, 绝缘子, 目标检测, 旋转矫正

Abstract: In this paper,a Gaussian Distance Intersection over Union(GDIoU) loss function based YOLOv4 deep learning method is proposed to solve the problem of low speed and low accuracy insulator positioning in the process of power line health inspection.In the scheme,a GDIoU loss function is designed to accelerate the convergence speed of the YOLOv4 deep learning network,and the two-dimensional Gaussian model is used to improve the convergence ability of the network,through which the perfor-mance of the YOLOv4 network is enhanced and the insulator's positioning accuracy is accordingly improved.At the same time,an adaptive tilt correction algorithm is proposed to improve the positioning accuracy of the insulators in different spatial angle states by rotating the image with only one insulator.The experimental results show that the average precision is increased by 7.37% compared with the naive YOLOv4 scheme.And the GDIoU based YOLOv4 deep learning network combined with the adaptive tile correction method accelerates the insulator positioning speed by three times compared with the other insulator positioning me-thods at the same level of accuracy.The proposed method makes a good balance between accuracy and speed,and its performance can meet the requirement of online insulator positioning adequately.

Key words: Electric power check, GDIoU Loss, Insulator, Objective detection, Rotation correction

中图分类号: 

  • TP389.1
[1] YAN T T,YANG G D,YU J Z.Feature fusion based insulator detection for aerial inspection[C]//2017 36th Chinese Control Conference(CCC).IEEE,2017:10972-10977.
[2] ZHOU S P,LEI X,QIAO B C,et al.Research on insulator fault diagnosis and remote monitoring system based on infrared images[J].Procedia Computer Science,2017,109:1194-1199.
[3] FANG T,JIN X,HU X L,et al.A Fast Insulator-Contour-Detection-Algorithm on Power Transmission Lines Images[C]//Applied Mechanics & Materials.2012:337-343.
[4] REDDY M J B,CHANDRA B K,MOHANTA D K.Condition Monitoring of 11 kV Distribution System Insulators Incorporating Complex Imagery Using Combined DOST-SVM Approach[J].IEEE Transactions on Dielectrics and Electrical Insulation,2013,20(2):664-74.
[5] ZOU Z X,SHI Z W,GUO Y H,et al.Object detection in 20years:A survey[J].arXiv:1905.05055,2019.
[6] ZHAO Z P,XU G Z,QI Y C,et al.Multi-patch deep features for power line insulator status classification from aerial images[C]//2016 International Joint Conference on Neural Networks(IJCNN).IEEE,2016:3187-3194.
[7] LIAO S L,AN J B.A Robust Insulator Detection Algorithm Based on Local Features and Spatial Orders for Aerial Images[J].IEEE Geoscience and Remote Sensing Letters,2015,12(5):963-967.
[8] PRATES R M,CRUZ R,MAROTTA A P,et al.Insulator visual non-conformity detection in overhead power distribution lines using deep learning[J].Computers & Electrical Engineering,2019,78:343-55.
[9] LEU X S,SUI Z H.Intelligent fault detection of high voltage line based on the Faster R-CNN[J].Measurement,2019,138:379-85.
[10] ZHAO Z B,ZHEN Z,ZHANG L,et al.Insulator DetectionMethod in Inspection Image Based on Improved Faster R-CNN[J].Energies,2019,12(7):1204.
[11] TAO X,ZHANG D,WANG Z,et al.Detection of power line insulator defects using aerial images analyzed with convolutional neural networks[J].IEEE Transactions on Systems,Man,and Cybernetics:Systems,2018,50(4):1486-1498.
[12] BOCHKOVSKIY A,WANG C Y,LIAO H Y M.YOLOv4:Optimal Speed and Accuracy of Object Detection[J].arXiv:2004.10934,2020.
[13] REDMON J,DIVVALA S,GIRSHICK R,et al.You only look once:Unified,real-time object detection[C]//IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).2016:779-788.
[14] REDMON J,FARHADDI A.YOLO9000:better,faster,stron-ger[C]//Proceedings of the IEEEConference on Computer Vision and Pattern Recognition(CVPR).2017:7263-7271.
[15] REDMON J,FARHADI A.Yolov3:An incremental improve-ment[J].arXiv:1804.02767,2018.
[16] WANG C Y,LIAO H Y,WU Y H,et al.CSPNet:A New Backbone that can Enhance Learning Capability of CNN[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops(CVPRW).2020:390-391.
[17] LIU S,QI L,QIN H F,et al.Path Aggregation Network for Instance Segmentation[C]//IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).2018:8759-8768.
[18] HE K M,ZHANG X Y,REN S Q,et al.Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2015,37(9):1904-1916.
[19] ZHENG Z H,WANG P,LIU W,et al.Distance-IoU Loss:Faster and Better Learning for Bounding Box Regression[J].arXiv:1911.08287,2019.
[20] LIN T Y,MAIRE M,BELONGIE S,et al.Microsoft COCO:Common Objects in Context[C]//European Conference on Computer Vision.2014:740-755.
[21] WANG T C,ANWER R M,CHOLAKKAL H,et al.Learning Rich Features at High-Speed for Single-Shot Object Detection[C]//2019 IEEE/CVF International Conference on Computer Vision(ICCV).2020:1971-1980.
[22] LIU W,ANGUELOV D,ERHAN D,et al.SSD:Single ShotMultiBox Detector[C]//European Conference on Computer Vision.Cham:Springer,2016:21-37.
[23] LIN T Y,GOYAL P,GIRSHICK R,et al.Focal Loss for Dense Object Detection[C]//Proceedings of the IEEE International Conference on Computer Vision.2017:2980-2988.
[24] TAN M X,PANG R M,LE Q V.EfficientDet:Scalable and Efficient Object Detection[C]//IEEE/CVF Conference on Compu-ter Vision and Pattern Recognition(CVPR).2020:10781-10790.
[25] REZATOFIGHI H,TSOI N,GWAK J Y,et al.Generalized Intersection over Union:A Metric and A Loss for Bounding Box Regression[C]//IEEE/CVF Conference on Computer Vision and Pattern Recognition(CVPR).2019:658-666.
[26] LING Z N,ZHANG D X,QIU R C,et al.An Accurate and Real-time Method of Self-blast Glass Insulator Location Based on Faster R-CNN and U-net with Aerial Images[J].CSEE Journal of Power and Energy Systems,2019,5(4):474-482.
[27] LIU C Y,WU Y Q,LIU J J,et al.MTI-YOLO:A Light-Weight and Real-Time Deep Neural Network for Insulator Detection in Complex Aerial Images[J].Energies,2021,14(5):1426.
[1] 刘冬梅, 徐洋, 吴泽彬, 刘倩, 宋斌, 韦志辉.
基于边框距离度量的增量目标检测方法
Incremental Object Detection Method Based on Border Distance Measurement
计算机科学, 2022, 49(8): 136-142. https://doi.org/10.11896/jsjkx.220100132
[2] 王灿, 刘永坚, 解庆, 马艳春.
基于软标签和样本权重优化的Anchor Free目标检测算法
Anchor Free Object Detection Algorithm Based on Soft Label and Sample Weight Optimization
计算机科学, 2022, 49(8): 157-164. https://doi.org/10.11896/jsjkx.210600240
[3] 祝文韬, 兰先超, 罗唤霖, 岳彬, 汪洋.
改进Faster R-CNN的光学遥感飞机目标检测
Remote Sensing Aircraft Target Detection Based on Improved Faster R-CNN
计算机科学, 2022, 49(6A): 378-383. https://doi.org/10.11896/jsjkx.210300121
[4] 陈永平, 朱建清, 谢懿, 吴含笑, 曾焕强.
基于外接圆半径差损失的实时安全帽检测算法
Real-time Helmet Detection Algorithm Based on Circumcircle Radius Difference Loss
计算机科学, 2022, 49(6A): 424-428. https://doi.org/10.11896/jsjkx.220100252
[5] 陈佳舟, 赵熠波, 徐阳辉, 马骥, 金灵枫, 秦绪佳.
三维城市场景中的小物体检测
Small Object Detection in 3D Urban Scenes
计算机科学, 2022, 49(6): 238-244. https://doi.org/10.11896/jsjkx.210400174
[6] 李发光, 伊力哈木·亚尔买买提.
基于改进CenterNet的航拍绝缘子缺陷实时检测模型
Real-time Detection Model of Insulator Defect Based on Improved CenterNet
计算机科学, 2022, 49(5): 84-91. https://doi.org/10.11896/jsjkx.210400142
[7] 胡伏原, 万新军, 沈鸣飞, 徐江浪, 姚睿, 陶重犇.
深度卷积神经网络图像实例分割方法研究进展
Survey Progress on Image Instance Segmentation Methods of Deep Convolutional Neural Network
计算机科学, 2022, 49(5): 10-24. https://doi.org/10.11896/jsjkx.210200038
[8] 徐涛, 陈奕仁, 吕宗磊.
基于改进YOLOv3的机坪工作人员反光背心检测研究
Study on Reflective Vest Detection for Apron Workers Based on Improved YOLOv3 Algorithm
计算机科学, 2022, 49(4): 239-246. https://doi.org/10.11896/jsjkx.210200119
[9] 张侣, 周博文, 吴亮红.
基于改进卷积注意力模块与残差结构的SSD网络
SSD Network Based on Improved Convolutional Attention Module and Residual Structure
计算机科学, 2022, 49(3): 211-217. https://doi.org/10.11896/jsjkx.201200019
[10] 赫晓慧, 邱芳冰, 程淅杰, 田智慧, 周广胜.
基于边缘特征融合的高分影像建筑物目标检测
High-resolution Image Building Target Detection Based on Edge Feature Fusion
计算机科学, 2021, 48(9): 140-145. https://doi.org/10.11896/jsjkx.200800002
[11] 袁磊, 刘紫燕, 朱明成, 马珊珊, 陈霖周廷.
融合改进密集连接和分布排序损失的遥感图像检测
Improved YOLOv3 Remote Sensing Target Detection Based on Improved Dense Connection and Distributional Ranking Loss
计算机科学, 2021, 48(9): 168-173. https://doi.org/10.11896/jsjkx.200800001
[12] 龚浩田, 张萌.
基于关键点检测的无锚框轻量级目标检测算法
Lightweight Anchor-free Object Detection Algorithm Based on Keypoint Detection
计算机科学, 2021, 48(8): 106-110. https://doi.org/10.11896/jsjkx.200700161
[13] 李琳, 刘学亮, 赵烨, 纪平.
结合乐高滤波器和SSD的低光照图像融合检测方法
Low Light Image Fusion Detection Method Based on Lego Filter and SSD
计算机科学, 2021, 48(7): 213-218. https://doi.org/10.11896/jsjkx.200800127
[14] 辛元雪, 史朋飞, 薛瑞阳.
基于区域提取与改进 LBP 特征的运动目标检测
Moving Object Detection Based on Region Extraction and Improved LBP Features
计算机科学, 2021, 48(7): 233-237. https://doi.org/10.11896/jsjkx.200600131
[15] 张曼, 李杰, 朱新忠, 沈霁, 成昊天.
基于改进DCGAN算法的遥感数据集增广方法
Augmentation Technology of Remote Sensing Dataset Based on Improved DCGAN Algorithm
计算机科学, 2021, 48(6A): 80-84. https://doi.org/10.11896/jsjkx.200700185
Viewed
Full text


Abstract

Cited

  Shared   
  Discussed   
No Suggested Reading articles found!