uaggan

A Pytorch implementation of "Unsupervised Attention-Guided Image-to-Image Translation", NIPS 2018, [Paper] | [TF code]

Installation

For Conda users, we provide a installation script ./scripts/conda_deps.sh. Alternatively, you can create a new Conda environment using this command:

conda env create -f environment.yml

Prepare Dataset:

Download a CycleGAN dataset (e.g. horse2zebra, apple2orange):

bash ./datasets/download_cyclegan_dataset.sh horse2zebra

Use your own dataset: put images from each domain at folder_a and folder_b respectively.

Usage

Training

sh ./scripts/train_uaggan.sh <gpu_id>

nohup sh -u ./scripts/train_uaggan.sh <gpu_id> > uag.log &

Test

sh ./scripts/test_uaggan.sh <gpu_id>

Mapping results

horse2zebra (without early stopping, threshold=0.1)

Real A	Attention Mask	Fake B	Real B	Attention Mask	Fake A

[More Results >>>]

horse2zebra (with early stopping, threshold=0.1)

//TODO

apple2orange

//TODO

Faster version

Combining the attention module and domain translation module into one generator, inspired by GANimation. (Results are not good.)

Training

sh ./scripts/train_faster_uaggan.sh <gpu_id>

Testing

sh ./scripts/test_faster_uaggan.sh <gpu_id>

Real A	Attention Mask	Fake B	Real B	Attention Mask	Fake A

Acknowledgment

This code is based on the pytorch-CycleGAN-and-pix2pix. Thanks to the contributors of this project.

yhlleo/uaggan