Lightweight Insulator and Defect Detection Method Based on Improved YOLOv8

Abstract

Insulator and defect detection is a critical technology for the automated inspection of transmission and distribution lines within smart grids. However, the development of a lightweight, real-time detection platform suitable for deployment on drones faces significant challenges. These include the high complexity of existing algorithms, limited availability of UAV images, and persistent issues with false positives and missed detections. To address this issue, this paper proposed a lightweight drone-based insulator defect detection method (LDIDD) that integrates data augmentation and attention mechanisms based on YOLOv8. Firstly, to address the limitations of the existing insulator dataset, data augmentation techniques are developed to enhance the diversity and quantity of samples in the dataset. Secondly, to address the issue of the network model’s complexity hindering its application on UAV equipment, depthwise separable convolution is incorporated for lightweight enhancement within the YOLOv8 algorithm framework. Thirdly, a convolutional block attention mechanism is integrated into the feature extraction module to enhance the detection of small insulator targets in aerial images. The experimental results show that the improved network reduces the computational volume by 46.6% and the mAP stably maintains at 98.3% compared to YOLOv8, which enables the implementation of a lightweight insulator defect network suitable for the UAV equipment side without affecting the detection performance.

Keywords

• insulator detection
• YOLOv8
• data augmentation
• lightweight network
• attention mechanism