Chinese Journal of Intelligent Science and Technology ›› 2021, Vol. 3 ›› Issue (3): 312-321.doi: 10.11959/j.issn.2096-6652.202132

• Special Issue: Intelligent Object Detection and Recognition • Previous Articles     Next Articles

Insulator detection based on SE-YOLOv5s

Qing TIAN1, Rong HU1, Zuoyong LI2,3, Yuanzheng CAI2,3, Zhaochai YU2,3   

  1. 1 College of Computer Science and Mathematics, Fujian University of Technology, Fuzhou 350118, China
    2 College of Computer and Control Engineering, Minjiang University, Fuzhou 350121, China
    3 Fujian Provincial Key Laboratory of Information Processing and Intelligent Control, Fuzhou 350121, China
  • Revised:2021-07-27 Online:2021-09-15 Published:2021-09-01
  • Supported by:
    The Natural Science Foundation of Fujian Province(2020J02024);The Natural Science Foundation of Fujian Province(2020J01828);Fuzhou Science and Technology Project(2020-RC-186)

Abstract:

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 application 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 attention 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 function 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.

Key words: insulator, object detection, deep learning, YOLOv5s

CLC Number: 

No Suggested Reading articles found!