Insulator defect detection algorithm based on adaptive feature fusion and lightweight YOLOv5s
Zhendong He,Yiming Wang,Anping Zheng,Jie Liu,Taishan Lou,Jie Zhang,Penghao Jiang,Jiong Xu
DOI: https://doi.org/10.1007/s11554-024-01589-4
IF: 2.293
2024-12-01
Journal of Real-Time Image Processing
Abstract:In power line inspections, detecting insulator defects is critical due to the potential breakdown and damage resulting from long-term exposure to the natural environment. However, challenges persist in practical detection processes, such as the limited availability of insulator defect images, difficulty detecting minor defects, and high memory consumption of deep learning models. We propose an insulator defect detection algorithm based on adaptive feature fusion and the lightweight YOLOv5s model to address these issues. First, we enhance the existing dataset by employing techniques like flipping, random pixel manipulation, noise addition, and contrast/brightness adjustments to increase the diversity of training samples. Next, we integrate the adaptive feature fusion (ASFF) module to enable the network to learn relationships between different feature maps, enhancing semantic information and improving the network's ability to detect minor defects. Finally, we replace the backbone of YOLOv5s with the lightweight convolutional neural network EfficientNet, making the network more efficient and enabling it to focus more on target features. Experimental results demonstrate significant improvements in insulator defect detection. The model achieves an accuracy rate of 96.4%, a recall rate of 96.1%, and an mAP of 97.6%, effectively enhancing network performance.
computer science, artificial intelligence,engineering, electrical & electronic,imaging science & photographic technology