Data augmentation method for insulators based on Cycle GAN

Abstract

Data augmentation is an important task of using existing data to expand data sets. Using generative countermeasure network technology to realize data augmentation has the advantages of high-quality generated samples, simple training, and fewer restrictions on the number of generated samples. However, in the field of transmission line insulator images, the freely synthesized samples are prone to produce fuzzy backgrounds and disordered samples of the main insulator features. To solve the above problems, this paper uses cycle generative adversarial network (Cycle-GAN) used for domain conversion in the generation countermeasure network as the initial framework and uses the self-attention mechanism and the channel attention mechanism to assist the conversion to realize the mutual conversion of different insulator samples. The attention module with prior knowledge is used to build the generation countermeasure network, and the GAN model with local controllable generation is built to realize the directional generation of insulator belt defect samples. The experimental results show that the samples obtained by this method are improved in a number of quality indicators, and the quality effect of the samples obtained is excellent, which has a reference value for the data expansion of insulator images.