CAD2Render: A Synthetic Data Generator for Training Object Detection and Pose Estimation Models in Industrial Environments

Abstract

Computer vision systems become more wide spread in the manufacturing industry for automating tasks. As these vision systems use more and more machine learning opposed to the classic vision algorithms, streamlining the process of creating the training datasets become more important. Creating large labeled datasets is a tedious and time consuming process that makes it expensive. Especially in a low-volume high-variance manufacturing environment. To reduce the costs of creating training datasets we introduce CAD2Render, a GPU-accelerated synthetic data generator based on the Unity High Definition Render Pipeline * This is an extended and substantially revised version of the paper "CAD2Render: A Modular Toolkit for GPU-Accelerated Moonen, et al. (HDRP). CAD2Render streamlines the process of creating highly customizable synthetic datasets with a modular design for a wide range of variation settings. We validate our toolkit by showcasing the performance of AI vision models trained purely with synthetic data. The performance is tested on object detection and pose estimation problems in a variate of industrial relevant use cases. The code for CAD2Render is available at https: //github.com/EDM-Research/CAD2Render