Lua
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
images
LICENSE
README.md
configuration.lua
dataset.lua
debugger.lua
helper_functions.lua
model_and_criterion.lua
plotting_and_image_creation.lua
train_ss-infogan.lua

README.md

Semi-supervised infoGAN

Official Torch Code for the paper Guiding InfoGAN with Semi-Supervision.

The script implements the training routine for the MNIST dataset. We used a cropped version of size 28x28 which can be found here.

After cloning this repository, download the dataset into the code directory and specify the path to it in the configuration.lua file, alongside any other parameter changes you wish to have. Then execute the training procedure in train_ss-infogan.lua with

th train_ss-infogan.lua

If the score with respect to a classifier is required, a valid classifier must be trained. Its path can then be specified in the configuration.lua file. If none is available, the metric will not be calculated.

Examples

MNIST

MNIST 1 MNIST 2

SVHN

SVHN 1 SVHN 2

CelebA

CELEBA 1 CELEBA 2

CIFAR-10

CIFAR-10

Dependencies

luarocks install torchnet
luarocks install penlight
luarocks install nninit
luarocks install cutorch
luarocks install cunn

Credits

This code is a heavily modified version of and based on the code found here.