• Stars
    star
    265
  • Rank 154,577 (Top 4 %)
  • Language
    Lua
  • License
    MIT License
  • Created over 8 years ago
  • Updated about 7 years ago

Reviews

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

Repository Details

A torch implementation of "Pixel-Level Domain Transfer"

Pixel Level Domain Transfer

A torch implementation of "Pixel-Level Domain Transfer". based on dcgan.torch.

Dataset

The dataset used is "LookBook", from Donggeun Yoo.

Training

To train the model, put the LOOKBOOK dataset under repository, resize images to 64*64. Prepare the dataset using prepare_data.ipynb. Then run

th main.lua

You can tune the parameters, such as number of filters, optimizer, etc.

Example results

Example results on LOOKBOOK dataset(top), left is input, right is generated clothes. Results on a similar dataset (bottom). More results will be added soon.

Results

Monitor the performance

  • Install display package with: luarocks install https://raw.githubusercontent.com/szym/display/master/display-scm-0.rockspec
  • Start the server with: th -ldisplay.start
  • Open this URL in your browser: http://localhost:8000

Below shows the results after 7 epochs, each 3*1 block is generated cloth, true cloth, input image. Errors of G, D, and A network will be plotted.

epoch 7