GithubHelp home page GithubHelp logo

gumpfly / cross-domain-disen Goto Github PK

View Code? Open in Web Editor NEW

This project forked from agonzgarc/cross-domain-disen

0.0 1.0 0.0 3.85 MB

Cross-domain disentanglement network

License: Other

Python 100.00%

cross-domain-disen's Introduction

Cross-domain disentantanglement network

Code for the paper "Image-to-image translation for cross-domain disentanglement", NeurIPS 2018.

Cross-domain disentanglement network

Based on this pix2pix implementation by Christopher Hesse, extensively explained in this article.

Setup

Please follow the setup described here. Tested with Tensorflow 1.8.0.

See DATA/MNISTCDCB/ for example images of our MNIST-CD/CB dataset.

Usage

In order to train a MODEL using DATA, run

python run_cross_domain_disen.py \
  --mode train \ 
  --output_dir checkpoints/MODEL \ 
  --input_dir DATA/train/  

Once the model finished training, it can be tested by running

python run_cross_domain_disen.py \ 
  --mode test \ 
  --output_dir test/MODEL \
  --checkpoint checkpoints/MODEL \
  --input_dir DATA/test/  

In order to extract disentangled features for other tasks (e.g. cross-domain retrieval), run

python run_cross_domain_disen.py \ 
  --mode features \ 
  --output_dir features/MODEL \ 
  --checkpoint checkpoints/MODEL \ 
  --input_dir DATA/test/  

Pre-trained models

MNIST-CD/CB (w/ GRL)

MNIST-CD/CB (w/o GRL)

Cars

Chairs

Facades

Maps

Dataset

MNIST-CD/CB

Citation

Please, cite the following paper if you use this code:

@inproceedings{gonzalez-garcia2018NeurIPS,
  title={Image-to-image translation for cross-domain disentanglement},
  author={Gonzalez-Garcia, Abel and van de Weijer, Joost and Bengio, Yoshua},
  booktile={NeurIPS},
  year={2018}
}

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.