Skip to content

A torch implementation of "Pixel-Level Domain Transfer"

License

Notifications You must be signed in to change notification settings

stoneshi-1999/PixelDTGAN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

A torch implementation of "Pixel-Level Domain Transfer"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Lua 81.1%
  • Jupyter Notebook 18.9%