• Stars
    star
    281
  • Rank 147,023 (Top 3 %)
  • Language
    Python
  • License
    MIT License
  • Created almost 8 years ago
  • Updated almost 6 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Tensorflow implementation of 3D Generative Adversarial Network.

tf-3dgan

license arXiv Tag

Tensorflow implementation of 3D Generative Adversarial Network.

This is a tensorflow implementation of the paper "Learning a Probabilistic Latent Space of Object Shapes via 3D Generative-Adversarial Modeling"

Blog Post with interactive volume plots

Requirements

  • tensorflow>=1.0
  • visdom>=1.0.1 (for mesh visualization)
  • scipy
  • scikit-image
  • stl (optional)

One-line installation

pip install scipy scikit-image stl visdom

Data

  • Download the training data from the 3D Shapenet website
  • Extract the zip and modify the path appropriately in dataIO.py

Usage

Launch visdom by running

python -m visdom.server

To train the model (visdom will show generated chairs after every 200 minibatches)

python 3dgan_mit_biasfree.py 0 <path_to_model_checkpoint>

To generate chairs

python 3dgan_mit_biasfree.py 1 <path_to_trained_model>

Some sample generated chairs

Source code files

File Description
3dgan_mit_biasfree.py 3dgan as mentioned in the paper, with same hyperparams.
3dgan.py baseline 3dgan with fully connected layer at end of discriminator.
3dgan_mit.py 3dgan as mentioned in the paper with bias in convolutional layers.
3dgan_autoencoder.py 3dgan with support for autoencoder based pre-training.
3dgan_feature_matching.py 3dgan with additional loss of feature mathcing of last layers.
dataIO.py data input output and plotting utilities.
utils.py tensorflow utils like leaky_relu and batch_norm layer.

Todo

  • Host the trained models
  • Add argparser based interface
  • Add threaded dataloader
  • Release the pytorch and keras versions of the GAN.
  • Train for longer number of epochs to improve quality of generated chairs.

Contributors

  • @meetshah1995
  • @khushhallchandra

More Repositories

1

pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch
Python
3,323
star
2

mit-ocw-dl

Download all video lectures from a MIT-OCW course with a single command.
Python
101
star
3

crnn-music-genre-classification

Tensorflow Implementation of Convolutional Recurrent Neural Networks for Music Genre Classification
Python
52
star
4

ConvolutionalNeuralNetwork

Convolutional Neural Networks for Object Classification done as a course project for CS663 Digital Image Processing
MATLAB
23
star
5

CS-663

Assignment Codes for CS663 Digital Image Processing
HTML
17
star
6

EE-702

Project codes for EE702 Computer Vision.
Python
12
star
7

CS-736

Assignment codes for CS736 Algorithms for Medical Image Processing.
HTML
10
star
8

config

My dotfiles
Python
9
star
9

TechFest2k15

Android Application Source for IIT-Bombay's TechFest 2015.
Java
7
star
10

CS-747

Assignment codes for CS747 Intelligent and Learning Agents
Python
5
star
11

CS-725

Assignment solutions for CS725 Machine Learning
Python
5
star
12

CS-229

Assignment Solutions to Coursera CS-229 Machine Learning by Stanford
MATLAB
5
star
13

glustup

Tools and Instruction Sets for GPU cluster setup.
4
star
14

ME-766

Assignment Solutions to course ME766 High Performance Scientific Computing.
C++
3
star
15

CS-749

Assignments for Course Digital Geometry Processing
C++
3
star
16

statistical-computing-interface

R based Statistical Computing Interface for easy statistical analysis and data visualization
JavaScript
3
star
17

EE-352

Solution codes for EE352 Digital Signal Processing Lab.
C
2
star
18

EE-340

Laboratory Files for course EE 340 Communications Lab
Python
2
star
19

EE-214

VHDL and Verilog Codes for Digital Lab.
VHDL
2
star
20

homepage

My Personal Webpage | Old, Redundant, Dirty and now depreciated.
CSS
2
star
21

EE-779

Assignment codes for EE779 Advanced Topics in Signal Processing
HTML
2
star
22

marching-cube-segmentation

Brain tumor segmentation using modified Marching Cubes segmentation algorithm
C++
2
star
23

EE-717

Assignment Solutions of Course EE717 Advanced Computing for Electrical Engineers
C++
2
star
24

fpmath

Automatically exported from code.google.com/p/fpmath
C++
2
star
25

CS-416

Assignment Solutions for Course Computer Network and Security.
HTML
2
star
26

auv_website

Website of Autonomous Underwater Vehicle Team IIT-Bombay
CSS
1
star
27

tf-devanagri

Convolutional Neural Networks to identify Devanagri characters implemented in Tensorflow
Python
1
star
28

sudoku-solver

Sudoku Solver written in lunch-break !
C++
1
star
29

EE-337

Assembly Codes for Microprocessors Laboratory
Assembly
1
star
30

EE-324

Solution codes and simulation files for EE324 Control Lab.
C
1
star
31

GMap-Fest-Scheduler

Generic Google Map based scheduling/routing framework that can be incorporated into any event or college fest Android app.
Java
1
star
32

dockerfiles

My public Dockerfiles
1
star
33

Vishynoid

Chess Robot ( BeagleBone )
C++
1
star
34

EE-338

Filter Design Assignment codes for course EE338 Digital Signal Processing
HTML
1
star
35

UFLDL-solutions

Programming Exercises Solutions for the Stanford Unsupervised Feature Learning and Deep Learning Tutorial
MATLAB
1
star
36

CS-213

Assignment Solutions for Course Data Structures and Algorithms..
C++
1
star