• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (03): 509-515.

• Artificial Intelligence and Data Mining • Previous Articles     Next Articles

A CycleGAN small sample library amplification method for faulty insulator detection

CUI Ke-bin,PAN Feng#br#   

  1. (School of Control and Computer Engineering,North China Electrical Power University,Baoding 071003,China)
  • Received:2020-04-29 Revised:2020-10-24 Accepted:2022-03-25 Online:2022-03-25 Published:2022-03-24

Abstract: In deep learning training, insulator detection requires a large number of faulty insulators. It is actually difficult to obtain a large amount of faulty insulator data. Generative Adversarial Network (GAN) provides a feasible solution for augmenting training samples. This paper supplements the defective insulator samples in the structure of the Cycle-consistent GAN (CycleGAN), optimizes the model by changing the loss function, and inputs the image synthesized by the forward generator to the reverse generator, thus maintaining the overall outline of the sample while adding the difference. In the SSD (Single Shot Detector) target detection experiment, the method of using the improved CycleGAN model to expand the dataset was compared with other GAN models. The results show that the method of using the improved CycleGAN to expand the dataset significantly improves the recognition rate of insulator drop detection compared with other expansion methods. 

Key words: cycle-consistent generative adversarial network, insulator, data augmentation, style transfer