Synthetic Data Generation Based on RDB-CycleGAN for Industrial Object Detection

Jiwei Hu, Feng Xiao, Qiwen Jin (), Guangpeng Zhao and Ping Lou
Additional contact information
Jiwei Hu: School of Information Engineering, Wuhan University of Technology, Wuhan 430070, China
Feng Xiao: School of Information Engineering, Wuhan University of Technology, Wuhan 430070, China
Qiwen Jin: School of Information Engineering, Wuhan University of Technology, Wuhan 430070, China
Guangpeng Zhao: School of Information Engineering, Wuhan University of Technology, Wuhan 430070, China
Ping Lou: School of Information Engineering, Wuhan University of Technology, Wuhan 430070, China

Mathematics, 2023, vol. 11, issue 22, 1-18

Abstract: Deep learning-based methods have demonstrated remarkable success in object detection tasks when abundant training data are available. However, in the industrial domain, acquiring a sufficient amount of training data has been a challenge. Currently, many synthetic datasets are created using 3D modeling software, which can simulate real-world scenarios and objects but often cannot achieve complete accuracy and realism. In this paper, we propose a synthetic data generation framework for industrial object detection tasks based on image-to-image translation. To address the issue of low image quality that can arise during the image translation process, we have replaced the original feature extraction module with the Residual Dense Block (RDB) module. We employ the RDB-CycleGAN network to transform CAD models into realistic images. Additionally, we have introduced the SSIM loss function to strengthen the network constraints of the generator and conducted a quantitative analysis of the improved RDB-CycleGAN-generated synthetic data. To evaluate the effectiveness of our proposed method, the synthetic data we generate effectively enhance the performance of object detection algorithms on real images. Compared to using CAD models directly, synthetic data adapt better to real-world scenarios and improve the model’s generalization ability.

Keywords: synthetic data; RDB-CycleGAN; image translation; object detection (search for similar items in EconPapers)
JEL-codes: C (search for similar items in EconPapers)
Date: 2023
References: View complete reference list from CitEc
Citations:

Downloads: (external link)
https://www.mdpi.com/2227-7390/11/22/4588/pdf (application/pdf)
https://www.mdpi.com/2227-7390/11/22/4588/ (text/html)

Related works:
This item may be available elsewhere in EconPapers: Search for items with the same title.

Export reference: BibTeX RIS (EndNote, ProCite, RefMan) HTML/Text

Persistent link: https://EconPapers.repec.org/RePEc:gam:jmathe:v:11:y:2023:i:22:p:4588-:d:1276926

Access Statistics for this article

Mathematics is currently edited by Ms. Emma He

More articles in Mathematics from MDPI
Bibliographic data for series maintained by MDPI Indexing Manager ().