Improved YOLOv5 foreign object detection for transmission lines

Abstract:

The traditional transmission line detection has the problems of low efficiency. To improve the performance, this paper proposes an improved you only look once version 5 (YOLOv5) transmission line foreign object detection algorithm. First, efficient channel attention (ECA) module is introduced in the backbone network for focusing the target features and improving the feature extraction capability of the network. Secondly, bilinear interpolation upsampling is introduced in the neck network to improve the model detection accuracy. Finally, by integrating the efficient intersection over union (EIoU) loss function and Soft non-maximum suppression (Soft NMS) algorithm, the convergence speed of the model is accelerated while the detection effect of the model is enhanced. Relative to the original algorithm, the improved algorithm reduces the number of parameters by 16.4%, increases the mean average precision (mAP)@0.5 by 3.9%, mAP@0.5:0.95 by 6.3%, and increases the detection speed to 55.3 frames per second (FPS). The improved algorithm is able to improve the performance of the foreign object detection in transmission lines effectively.