/PyTorch-Voxel-VAE

PyTorch implementation of https://github.com/IsaacGuan/3D-VAE with some modifications.

Primary LanguagePythonMIT LicenseMIT

PyTorch-Voxel-VAE

PyTorch implementation of https://github.com/IsaacGuan/3D-VAE with some modifications.

Left: Reconstruction, Right: Ground truth

Train

python3 train.py

Test

python3 test.py

Results are saved to ./reconstructions/