radeenxalnw / attention-cyclegan--van-gogh Goto Github PK

AttentionGAN Training/Testing

Download a dataset using the previous script (e.g., horse2zebra).

Test the Pretrained Model with pretrained file

Test the Trained Model python test.py --dataroot ./datasets/selfie2anime/ --name selfie2anime_attentiongan --model attention_gan --dataset_mode unaligned --norm instance --phase test --no_dropout --load_size 256 --crop_size 256 --batch_size 1 --gpu_ids 0 --num_test 5000 --epoch latest

@article{tang2021attentiongan, title={AttentionGAN: Unpaired Image-to-Image Translation using Attention-Guided Generative Adversarial Networks}, author={Tang, Hao and Liu, Hong and Xu, Dan and Torr, Philip HS and Sebe, Nicu}, journal={IEEE Transactions on Neural Networks and Learning Systems (TNNLS)}, year={2021} }

@inproceedings{tang2019attention, title={Attention-Guided Generative Adversarial Networks for Unsupervised Image-to-Image Translation}, author={Tang, Hao and Xu, Dan and Sebe, Nicu and Yan, Yan}, booktitle={International Joint Conference on Neural Networks (IJCNN)}, year={2019} }

Some outputs are given below

Computer Generated Van Gogh

Reference :

1. https://arxiv.org/abs/1903.12296
2. https://github.com/victor369basu/CycleGAN-with-Self-Attention
3. https://github.com/Ha0Tang/AttentionGAN