Optics and Precision Engineering, Volume. 31, Issue 16, 2406(2023)

Equipment fault dataset amplification method combine 3D model with improved CycleGAN

Baoping LI... Hengyi QI*, Manli WANG and Po WEI |Show fewer author(s)
Author Affiliations
  • College of Physics and Electronic Information, Henan Polytechnic University, Jiaozuo454000, China
  • show less

    The performance of deep-learning-based equipment fault detection systems relies heavily on the size and class diversity of the sample set. Because it is difficult to collect all types of fault sample comprehensively in industrial production, there is a demand for sample set augmentation. A fault dataset amplification method combining 3D modeling with an improved cycle generative adversarial network (CycleGAN) is proposed. First, various equipment malfunction images generated by 3D modeling software are applied to the CycleGAN network training to guide it in generating pseudo-real images to address the problem of insufficient samples and an uneven distribution. Second, a U-ResNet generator is used in the CycleGAN network to solve the problem of edge blurring and gradient vanishing during network training. The method was applied to the task of belt conveyor deviation detection. The experimental results show that the contour structure of the method converges quickly in the training process and has good timeliness in comparison with other amplification methods. The accuracy rate of the method is 98.1% when applying to the target detection network, which is 4.5% higher than that of the original real dataset. It meets the basic requirements of a balanced distribution of amplified datasets and high image quality.

    Tools

    Get Citation

    Copy Citation Text

    Baoping LI, Hengyi QI, Manli WANG, Po WEI. Equipment fault dataset amplification method combine 3D model with improved CycleGAN[J]. Optics and Precision Engineering, 2023, 31(16): 2406

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Information Sciences

    Received: Nov. 16, 2022

    Accepted: --

    Published Online: Sep. 5, 2023

    The Author Email: QI Hengyi (mystericq@home.hpu.edu.cn)

    DOI:10.37188/OPE.20233116.2406

    Topics