基于SE-YOLOv5s的绝缘子检测  

Insulator detection based on SE-YOLOv5s

作　　者：田庆 胡蓉[1] 李佐勇 蔡远征 余兆钗 TIAN Qing;HU Rong;LI Zuoyong;CAI Yuanzheng;YU Zhaochai(College of Computer Science and Mathematics,Fujian University of Technology,Fuzhou 350118,China;College of Computer and Control Engineering,Minjiang University,Fuzhou 350121,China;Fujian Provincial Key Laboratory of Information Processing and Intelligent Control,Fuzhou 350121,China)

机构地区：[1]福建工程学院计算机科学与数学学院,福建福州350118 [2]闽江学院计算机与控制工程学院,福建福州350121 [3]福建省信息处理与智能控制重点实验室,福建福州350121

出　　处：《智能科学与技术学报》2021年第3期312-321,共10页Chinese Journal of Intelligent Science and Technology

基　　金：福建省自然科学基金资助项目(No.2020J02024,No.2020J01828);福州市科技计划资助项目(No.2020-RC-186)。

摘　　要：在电力系统需要巡检的大环境下,人工巡检的传统方式存在很大不便和安全隐患,而采用无人机的目标检测方法在绝缘子检测识别方向有很大的应用前景。针对绝缘子图像检测中存在的场景复杂、视角多变、设备计算能力受限等问题,提出了一种改进的轻量级SE-YOLOv5s卷积神经网络来实现对绝缘子的快速目标检测,该方法首先在YOLOv5s网络中融合SE注意力模块,以强化网络对绝缘子目标的辨识能力,随后采用K-means聚类方法构建绝缘子的先验框,以提升定位精度,最后构造置信度与定位任务联合的损失函数,并引入Mosaic数据增强策略训练网络,有效解决训练数据不足的问题。经过实验验证发现,与主流目标检测方法相比,提出的SE-YOLOv5s方法在绝缘子检测准确率、召回率、检测速度及平均精度均值等性能指标上均取得了较好的结果。实验结果表明,该网络对于绝缘子检测有很好的效果,具有更好的鲁棒性,对电力系统的巡检方式具有参考价值。In a large environment where the power system needs to be inspected,the traditional method of manual inspection has great inconvenience and potential safety risks,and the object detection method of unmanned aerial vehicle has great ap-plication prospects in the direction of insulator detection and recognition.SE-YOLOv5s,a lightweight insulator detection network that performs efficient detection for this task was presented.Firstly,backbone of YOLOv5s by fusing the SE atten-tion module was strengthen.Then,the position distribution of insulator object was investigated and predefined templates of a priori box by K-means clustering on prior coordinate vectors were generated.Finally,the network by a multitask loss func-tion was trained combined with confidence and position regression task.Furthermore,Mosaic data augmentation was utilized to supplement additional training samples.Experimental results demonstrate that the proposed SE-YOLOv5s significantly outperforms baseline methods at multiple criterions including accuracy,recall rate,detection rate and mean average precision.In comparison with the baselines,the proposed network has a flexible trade-off between robustness and memory overhead and it is a potential approach to promote the power system development.

关 键 词：绝缘子 目标检测 深度学习 YOLOv5s 

分 类 号：TP391.41[自动化与计算机技术—计算机应用技术] TM216[自动化与计算机技术—计算机科学与技术]