https://github.com/LihangLiu/3d-colorization
https://github.com/LihangLiu/3d-colorization
3dshapenet colorization deeplearning tensorflow
Last synced: over 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/LihangLiu/3d-colorization
- Owner: LihangLiu
- Created: 2017-03-01T15:39:35.000Z (over 9 years ago)
- Default Branch: master
- Last Pushed: 2017-10-17T19:53:24.000Z (over 8 years ago)
- Last Synced: 2024-08-01T03:46:12.384Z (almost 2 years ago)
- Topics: 3dshapenet, colorization, deeplearning, tensorflow
- Language: Python
- Size: 17.6 MB
- Stars: 1
- Watchers: 3
- Forks: 1
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# 3D Colorization
This repository contains implementations of several algorithms for 3d volume colorization. Given a binary voxel, the model outputs the same shape with colorizations. To run the code, you should have the tensorflow installed in your python.
## Dataset
Our model takes the 3d voxel representations as input. We used Shapenet Core dataset for training. Before training, the conversion from obj format to voxel representations is required and implenmented under the 3d-colorization/objLoader directory. The dataset can't be provided directly due to license limitations. Please contact the Shapenet team for dataset downloading.
## Algorithms for 3d volume colorization
### cwgan
Conditional Wasserstein GAN. To run:
cd src64/cwgan
python train.py
### cwgan with surface kernel
Still in process. To run:
cd src64/cwgan-conv2_5d
python train.py
### vae
Variational Autoencoder. To run:
cd src-vae/vae
python train.py
### began
Boundary Equibilibrium GAN. To run:
cd src64/began
python train.py
### hourglass
Stacked Hourglass Networks. Only two stacks due to the meomory limitation issue. To run:
cd src-vae
python train.py