Pose-Transfer
PyTorch Implementation of Deformable GAN https://arxiv.org/abs/1801.00055
Also check out the original repo in Keras, by AliaksandrSiarohin - https://github.com/AliaksandrSiarohin/pose-gan
Baseline Model ( run in src_baseline )
Fasion :
CUDA_VISIBLE_DEVICES=1 python main.py --l1_penalty_weight 10 --batch_size 4 --number_of_epochs 90 --gen_type baseline --expID baseline_fasion --pose_dim 18 --dataset fasion
Human 3.6 :
CUDA_VISIBLE_DEVICES=0 python main.py --l1_penalty_weight 10 --batch_size 4 --number_of_epochs 90 --gen_type baseline --expID baseline_h36m --dataset h36m
Deformable Model ( run in src_deformable )
Fasion :
CUDA_VISIBLE_DEVICES=1 python main.py --warp_skip mask --l1_penalty_weight 0.01 --batch_size 2 --number_of_epochs 90 --gen_type baseline --expID full_fasion --pose_dim 18 --dataset fasion --nn_loss_area_size 5 --batch_size 2 --content_loss_layer block1_conv2
Human 3.6 :
CUDA_VISIBLE_DEVICES=1 python main.py --warp_skip mask --l1_penalty_weight 0.01 --batch_size 2 --number_of_epochs 90 --gen_type baseline --expID full_h36m --dataset h36m --nn_loss_area_size 5 --batch_size 2 --content_loss_layer block1_conv2
Specify data directory by passing data_Dir option.
ToDo :
-> Add code for loading annotations and train/test pairs for Fasion and Human 3.6 dataset.