Insulator defect detection based on improved you-only-look-once v4 in complex scenarios
Suzhen Wang, Hao Xu, Mingwei Shao, Lin Zhao · Journal of Electronic Imaging · 2023
As power grid structure becomes increasingly complex, the impact of the operation status of key substation equipment on the safe and stable operation of the power grid is gradually increasing. Aiming at the problems of dense targets and occlusion recognition difficulties in the complex background of substations, an improved you-only-look-once (YOLO) v4 insulator defect detection algorithm is proposed. A large number of insulator and defect datasets were collected from the Internet. We use MobileNet-v2 to replace the backbone network of YOLOv4, and in view of the problem that the accuracy of YOLOv4 target detection model is not high for small targets or dense targets, convolutional block attention module (CBAM) attention mechanism is added to the backbone network to enhance the feature extraction ability, and ultra-lightweight subspace attention module (ULSAM) is added to the detection head to learn the cross-channel information. Experiments based on InDataset show that the insulator defect detection algorithm based on deep learning proposed achieves a high detection accuracy and detection speed; the mean average precision of the improved YOLOv4 model is 87.48%, and the recall rate of insulator defects reaches 79.84%, which meets the requirements of robustness and accuracy of insulator defect detection in real scenarios.