Single Image 3D Reconstruction based on Conditional GAN [report]
Code for EECS 504: Introduction to Computer Vision Project
Team Members:
Danish Syed*, Hansal Shah*
Manas Buragohain*,
* Indicates equal contribution
- PyTorch 1.4.0
- PyTorch 3D
- CUDA 10.1
- cuDNN 7.6
Pixel2Mesh Baseline
python train_p2m.py --config-yml config/train_p2m.yml
Vanilla SICGAN
python train_p2m_gan.py --config-yml config/train_p2m_gan.yml
SICGAN with random noise
python train_p2m_randgan.py --config-yml config/train_p2m_randgan.yml
The SICGAN is realeased under MIT License