Optoelectronics Letters, Volume. 20, Issue 8, 490(2024)
Improved YOLOv5 foreign object detection for transmission lines
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.
Get Citation
Copy Citation Text
ZHOU Liming, LI Shixin, ZHU Zhiren, CHEN Fankai, LIU Chen, and DONG Xiuhuan. Improved YOLOv5 foreign object detection for transmission lines[J]. Optoelectronics Letters, 2024, 20(8): 490
Received: Oct. 15, 2023
Accepted: Apr. 9, 2024
Published Online: Aug. 23, 2024
The Author Email: Shixin LI (Li_shixin@sin.com)