Reproduction of Pose-Transfer-PATN

Paper: "Progressive Pose Attention Transfer for Person Image Generation"[pdf]

This project refers the official code link, if you find this project useful, please cite the offical paper.

@inproceedings{zhu2019progressive,
  title={Progressive Pose Attention Transfer for Person Image Generation},
  author={Zhu, Zhen and Huang, Tengteng and Shi, Baoguang and Yu, Miao and Wang, Bofei and Bai, Xiang},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  pages={2347--2356},
  year={2019}
}

用我的滴滴云AI大师码0003在滴滴云上购买GPU/vGPU/机器学习产品可享受9折优惠，点击前往滴滴云官网

Difference from Official Code

• Developed based on the pytorch template link
• Set hyperparameters using a configuration class
• Added warmup tricks

Results on Market1501 (SSIM)

Model	Market1501
ResNet50 (128x64)	0.311

Get Started

1. cd to folder where you want to download this repo

2. Run git clone https://github.com/lulujianjie/pose-transfer-PATN.git

3. Install dependencies:

   • pytorch>=0.4
   • torchvision
   • yacs
   • skimage
   • scipy

Train

python train.py

Test

python test.py