MU-GAN: Facial Attribute Editing based on Multi-attention Mechanism

Official implementation of MU-GAN

Requirements

*Python 3

*PyTorch 0.4.0

Train:

sudo CUDA_VISIBLE_DEVICES=x python3 train.py --img_size 128 --experiment_name 128_shortcut1_inject1_none --gpu --num_workers x --batch_size x

Test:

sudo CUDA_VISIBLE_DEVICES=x python3 test_x_x.py --experiment_name 128_shortcut1_inject1_none --test_int 1.0 --gpu --load_epoch x

Currently, the different variants of the code are in a mess. We will reorganize the code when we return to school.

This code refers to the projects:AttGAN-PyTorch Thanks for their excellent work！

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
__pycache__		__pycache__
useful_tool		useful_tool
LICENSE		LICENSE
README.md		README.md
attgan.py		attgan.py
attgan_1_0.py		attgan_1_0.py
attgan_2_0.py		attgan_2_0.py
attgan_3_0.py		attgan_3_0.py
attgan_3_0_1.py		attgan_3_0_1.py
attgan_4_2_1.py		attgan_4_2_1.py
attgan_5_1.py		attgan_5_1.py
attgan_5_2.py		attgan_5_2.py
attgan_5_5.py		attgan_5_5.py
attgan_final.py		attgan_final.py
data.py		data.py
helpers.py		helpers.py
my_att_list.txt		my_att_list.txt
nn.py		nn.py
read_txt.py		read_txt.py
readme.txt		readme.txt
requirements.txt		requirements.txt
switchable_norm.py		switchable_norm.py
test.py		test.py
test_multi_1_0.py		test_multi_1_0.py
test_slide.py		test_slide.py
train.py		train.py
utils.py		utils.py