Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (03): 509-515.
• Artificial Intelligence and Data Mining • Previous Articles Next Articles
CUI Ke-bin,PAN Feng#br#
Received:
Revised:
Accepted:
Online:
Published:
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
CUI Ke-bin, PAN Feng. A CycleGAN small sample library amplification method for faulty insulator detection[J]. Computer Engineering & Science, 2022, 44(03): 509-515.
0 / / Recommend
Add to citation manager EndNote|Ris|BibTeX
URL: http://joces.nudt.edu.cn/EN/
http://joces.nudt.edu.cn/EN/Y2022/V44/I03/509