CAD-based data augmentation and transfer learning empowers part classification in manufacturing
- Especially in manufacturing systems with small batches or customized products, as well as in remanufacturing and recycling facilities, there is a wide variety of part types that may be previously unseen. It is crucial to accurately identify these parts based on their type for traceability or sorting purposes. One approach that has shown promising results for this task is deep learning–based image classification, which can classify a part based on its visual appearance in camera images. However, this approach relies on large labeled datasets of real-world images, which can be challenging to obtain, especially for parts manufactured for the first time or whose appearance is unknown. To overcome this challenge, we propose generating highly realistic synthetic images based on photo-realistically rendered computer-aided design (CAD) data. Using this commonly available source, we aim to reduce the manual effort required for data generation and preparation and improve the classification performance of deep learning models using transfer learning. In this approach, we demonstrate the creation of a parametric rendering pipeline and show how it can be used to train models for a 30-class classification problem with typical engineering parts in an industrial use case. We also demonstrate how our method’s entropy gain improves the classification performance in various deep image classification models.
| Author: | Patrick Ruediger-FloreORCiD, Moritz Glatt, Marco Hussong, Jan C. Aurich |
|---|---|
| URN: | urn:nbn:de:hbz:386-kluedo-89243 |
| DOI: | https://doi.org/10.1007/s00170-023-10973-6 |
| ISSN: | 1433-3015 |
| Parent Title (English): | The International Journal of Advanced Manufacturing Technology |
| Publisher: | Springer Nature |
| Editor: | Andrew Yeh-Ching Nee, Kai Cheng, David W. Russel, M.S. Shunmugam, İsmail Lazoğlu |
| Document Type: | Article |
| Language of publication: | English |
| Date of Publication (online): | 2025/04/09 |
| Year of first Publication: | 2023 |
| Publishing Institution: | Rheinland-Pfälzische Technische Universität Kaiserslautern-Landau |
| Date of the Publication (Server): | 2025/04/16 |
| Issue: | (2023) Vol.125 |
| Page Number: | 14 |
| First Page: | 5605 |
| Last Page: | 5618 |
| Source: | https://link.springer.com/article/10.1007/s00170-023-10973-6 |
| Faculties / Organisational entities: | Kaiserslautern - Fachbereich Maschinenbau und Verfahrenstechnik |
| DDC-Cassification: | 6 Technik, Medizin, angewandte Wissenschaften / 670 Industrielle und handwerkliche Fertigung |
| Collections: | Open-Access-Publikationsfonds |
| Licence (German): |
