<p>An improved cycle-consistent generative adversarial network (CycleGAN) method for defect data augmentation based on feature fusion and self attention residual module is proposed to address the insufficiency of defect sample data for light guide plate (LGP) in production, as well as the problem of minor defects. Two optimizations are made to the generator of CycleGAN: fusion of low resolution features obtained from partial up-sampling and down-sampling with high-resolution features, combination of self attention mechanism with residual network structure to replace the original residual module. Qualitative and quantitative experiments were conducted to compare different data augmentation methods, and the results show that the defect images of the LGP generated by the improved network were more realistic, and the accuracy of the you only look once version 5 (YOLOv5) detection network for the LGP was improved by 5.6%, proving the effectiveness and accuracy of the proposed method.</p>

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Data augmentation method for light guide plate based on improved CycleGAN

  • Yefei Gong,
  • Chao Yan,
  • Ming Xiao,
  • Mingli Lu,
  • Hua Gao

摘要

An improved cycle-consistent generative adversarial network (CycleGAN) method for defect data augmentation based on feature fusion and self attention residual module is proposed to address the insufficiency of defect sample data for light guide plate (LGP) in production, as well as the problem of minor defects. Two optimizations are made to the generator of CycleGAN: fusion of low resolution features obtained from partial up-sampling and down-sampling with high-resolution features, combination of self attention mechanism with residual network structure to replace the original residual module. Qualitative and quantitative experiments were conducted to compare different data augmentation methods, and the results show that the defect images of the LGP generated by the improved network were more realistic, and the accuracy of the you only look once version 5 (YOLOv5) detection network for the LGP was improved by 5.6%, proving the effectiveness and accuracy of the proposed method.