Learning to Generalizing Kinematic Models to Novel Objects. [Paper] [Video]
To generate data, see this repository.
To train MDN on synthetic data, see magic/mixture/train_mdn.py
.
To run the script: python -m magic.mixture.train_model --name test_microwave --train-dir /data/nyi/local_ajinkya/datasets/screwNet/simulated_datasets/synArt/microwave --test-dir /data/nyi/local_ajinkya/datasets/screwNet/simulated_datasets/synArt/microwave-test --ntrain 160000 --ntest 16000 --cuda --ndof 1 --obj microwave