Research on improved YOLOv8 algorithm for insulator defect detection

Abstract

With the rapid advancement of artificial intelligence technologies, drone aerial photography has gradually become the mainstream method for defect detection of transmission line insulators. To address the issues of slow recognition speed and low accuracy in existing detection methods, this paper proposes an insulator defect detection algorithm based on an improved YOLOv8s model. Initially, a multi-scale large-kernel attention (MLKA) module is introduced to enhance the model’s focus on features of different scales as well as low-level feature maps. In addition, by employing lightweight GSConv convolution and constructing the GSC_C2f module, the computational process is simplified and memory burden is reduced, thereby effectively improving the performance of insulator defect detection. Finally, an improved loss function using SIoU is adopted to optimize the model’s detection performance and enhance its feature extraction capability for insulator defects. Experimental results demonstrate that the improved model exhibits excellent performance in drone aerial photography for insulator defect detection, achieving an mAP of 99.22% and an FPS of 55.73 frames per second. Compared to the original YOLOv8s and YOLOv5s, the improved model’s mAP increased by 2.18% and 2.91%, respectively, and the model size is only 30.18 MB, meeting the requirements for real-time operation and accuracy.