koryako / torch-gan Goto Github PK

To run the code clone the repository

git clone https://github.com/skaae/torch-gan.git

cd to the datasets subfolder and run create_dataset.py. This will create the labeled faces in the wildt dataset. This may take a while depending on your internet connection etc.

Then run

th train_lfw.lua -g 0

where -g 0 specifies the GPU you want to use. The code will only run on GPU, but you can easily modify to run on CPU by removing the cudnn dependencies.

The code will plot 100 generated images after each epoch. After a 5-10 epochs you should see something that looks like a face.

The code was written by Anders Boesen Lindbo Larsen and Søren Kaae Sønderby. Our code is based on code released with the LAPGAN paper.

Long latent space movie

• Torch
• numpy
• skimage
• h5py
• torch hdf5
• cudnn for torch