alokwhitewolf / Pytorch-Attention-Guided-CycleGAN

Pytorch implementation of Unsupervised Attention-guided Image-to-Image Translation.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Pytorch-Attention-Guided-CycleGAN

Attempt at Pytorch implementation of Unsupervised Attention-guided Image-to-Image Translation.

Image This architecture uses an attention module to identify the foreground or salient parts of the images onto which image translation is to be done.

Some of the results shown in the paper - Image

How to run

Download dataset

bash datasets/download_datasets.sh <cyclegan dataset arguement>

Train

train.py

optional arguments

--resume <checkpoint path to resume from>
--dataroot <root of the dataset images>
--LRgen <lr of generator>
--LRdis <lr of discriminator>
--LRattn <lr of attention module>

About

Pytorch implementation of Unsupervised Attention-guided Image-to-Image Translation.


Languages

Language:Python 95.0%Language:Shell 5.0%