├── .gitignore ├── DAMSMencoders └── .gitignore ├── LICENSE ├── README.md ├── code ├── .gitignore ├── GlobalAttention.py ├── cfg │ ├── DAMSM │ │ ├── bird.yml │ │ └── coco.yml │ ├── bird_attn2.yml │ ├── bird_attnDCGAN2.yml │ ├── coco_attn2.yml │ ├── eval_bird.yml │ ├── eval_bird_attnDCGAN2.yml │ └── eval_coco.yml ├── datasets.py ├── main.py ├── miscc │ ├── __init__.py │ ├── config.py │ ├── losses.py │ └── utils.py ├── model.py ├── pretrain_DAMSM.py └── trainer.py ├── data └── .gitignore ├── eval ├── FreeMono.ttf ├── GlobalAttention.py ├── README.md ├── __init__.py ├── data │ ├── bird_AttnGAN2.pth │ ├── captions.pickle │ └── text_encoder200.pth ├── dockerfile.cpu ├── dockerfile.gpu ├── eval.py ├── main.py ├── miscc │ ├── __init__.py │ ├── config.py │ └── utils.py ├── model.py └── requirements.txt ├── example_bird.png ├── example_coco.png ├── framework.png └── models └── .gitignore /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/.gitignore -------------------------------------------------------------------------------- /DAMSMencoders/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/README.md -------------------------------------------------------------------------------- /code/.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc 2 | !.gitignore -------------------------------------------------------------------------------- /code/GlobalAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/GlobalAttention.py -------------------------------------------------------------------------------- /code/cfg/DAMSM/bird.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/DAMSM/bird.yml -------------------------------------------------------------------------------- /code/cfg/DAMSM/coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/DAMSM/coco.yml -------------------------------------------------------------------------------- /code/cfg/bird_attn2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/bird_attn2.yml -------------------------------------------------------------------------------- /code/cfg/bird_attnDCGAN2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/bird_attnDCGAN2.yml -------------------------------------------------------------------------------- /code/cfg/coco_attn2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/coco_attn2.yml -------------------------------------------------------------------------------- /code/cfg/eval_bird.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/eval_bird.yml -------------------------------------------------------------------------------- /code/cfg/eval_bird_attnDCGAN2.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/eval_bird_attnDCGAN2.yml -------------------------------------------------------------------------------- /code/cfg/eval_coco.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/cfg/eval_coco.yml -------------------------------------------------------------------------------- /code/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/datasets.py -------------------------------------------------------------------------------- /code/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/main.py -------------------------------------------------------------------------------- /code/miscc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/miscc/__init__.py -------------------------------------------------------------------------------- /code/miscc/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/miscc/config.py -------------------------------------------------------------------------------- /code/miscc/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/miscc/losses.py -------------------------------------------------------------------------------- /code/miscc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/miscc/utils.py -------------------------------------------------------------------------------- /code/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/model.py -------------------------------------------------------------------------------- /code/pretrain_DAMSM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/pretrain_DAMSM.py -------------------------------------------------------------------------------- /code/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/code/trainer.py -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore 3 | -------------------------------------------------------------------------------- /eval/FreeMono.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/FreeMono.ttf -------------------------------------------------------------------------------- /eval/GlobalAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/GlobalAttention.py -------------------------------------------------------------------------------- /eval/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/README.md -------------------------------------------------------------------------------- /eval/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/__init__.py -------------------------------------------------------------------------------- /eval/data/bird_AttnGAN2.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/data/bird_AttnGAN2.pth -------------------------------------------------------------------------------- /eval/data/captions.pickle: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/data/captions.pickle -------------------------------------------------------------------------------- /eval/data/text_encoder200.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/data/text_encoder200.pth -------------------------------------------------------------------------------- /eval/dockerfile.cpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/dockerfile.cpu -------------------------------------------------------------------------------- /eval/dockerfile.gpu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/dockerfile.gpu -------------------------------------------------------------------------------- /eval/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/eval.py -------------------------------------------------------------------------------- /eval/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/main.py -------------------------------------------------------------------------------- /eval/miscc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/miscc/__init__.py -------------------------------------------------------------------------------- /eval/miscc/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/miscc/config.py -------------------------------------------------------------------------------- /eval/miscc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/miscc/utils.py -------------------------------------------------------------------------------- /eval/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/model.py -------------------------------------------------------------------------------- /eval/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/eval/requirements.txt -------------------------------------------------------------------------------- /example_bird.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/example_bird.png -------------------------------------------------------------------------------- /example_coco.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/example_coco.png -------------------------------------------------------------------------------- /framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/taoxugit/AttnGAN/HEAD/framework.png -------------------------------------------------------------------------------- /models/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !README.md 3 | !.gitignore --------------------------------------------------------------------------------