FINet: An Insulator Dataset and Detection Benchmark Based on Synthetic Fog and Improved YOLOv5

Zheng-De Zhang,Bo Zhang,Zhi-Cai Lan,Hai-Chun Liu,Dong-Ying Li,Ling Pei,Wen-Xian Yu
DOI: https://doi.org/10.1109/tim.2022.3194909
IF: 5.6
2022-10-05
IEEE Transactions on Instrumentation and Measurement
Abstract:The inspection of insulators and their defects is of great significance for ensuring the safety and stability of power system. Small sample is one of the main issues of insulator defect detection based on neural networks. In this research, we release a dataset for insulators and self-explosive defects detection and provide a benchmark based on improved YOLOv5, named Foggy Insulator Network (FINet). In this work, a synthetic fog algorithm is implemented and optimized. A synthetic foggy insulator dataset (SFID) with 13 000 images is constructed and released. The YOLOv5 network is improved into SE-YOLOv5 by introducing the channel attention mechanism, and a robust detection model with 96.2% F1 score for insulators and their defects is trained from scratch and served as a benchmark. The synthetic fog algorithm proposed in this article can be widely used for data augmentation of various datasets. The trained model can be applied in the field of transmission line inspection. The source codes, datasets, and tutorials are available on GitHub.
engineering, electrical & electronic,instruments & instrumentation
What problem does this paper attempt to address?