/detection_on_synth_data

Creating synthetic dataset in blender and training object detection model

Primary LanguagePython

Object detection on synthetic data

Creating synthetic dataset in blender and training object detection model

PytorchLightning Config: Hydra

Getting started

  1. Follow instructions to install Poetry: 
    # Unix/MacOs installation
curl -sSL https://install.python-poetry.org | python3 -
  2. Check that poetry was installed successfully: 
    poetry --version
  3. Setup workspace: 
    make setup_ws
  4. Setup ClearML: 
    clearml-init
  5. Download dataset to your local workspace: 
    make migrate_dataset

Training

make run_training

Inference

make get_weights
make run_inference

Results

Image 1 Image 2 Image 3