Counting 3,742 Big Data & Machine Learning Frameworks, Toolsets, and Examples...
Suggestion? Feedback? Tweet @stkim1

Unity Image Synthesis

This project will help you get up to speed with generating synthetic training images in Unity. You don't need any experience with Unity, but experience with Python and the fastai library/course is recommended. By the end of the tutorial, you will have trained an image segmentation network that can recognize different 3d solids.

Read more details on my blog

Follow along with the video tutorial

Getting Started

  1. Clone the repo
  2. Open the project with Unity / Unity Hub (2018.3.2 recommended)
  3. Open up the "Solids" scene
  4. Create a "captures/train" and "captures/val" folders
  5. Open up the SceneControl object and enable Save/grayscale if you want to do fastai training. Otherwise, leave those disabled and just see the annotations within Unity.
  6. Press the Run button
  7. Create a secondary Game window and change the Display to Display 3 for layer based category annotations.