conda create -n pise python=3.6
conda install pytorch=1.2 cudatoolkit=10.0 torchvision
pip install scikit-image pillow pandas tqdm dominate natsort
Data preparation for images and keypoints can follow Pose Transfer
Parsing data can be found from baidu (fectch code: abcd) or Google drive.
python train.py --name=fashion --model=painet --gpu_ids=0
You can directly download our test results from baidu (fetch code: abcd) or Google drive.
Pre-trained checkpoint reported in our paper can be found from baidu (fetch code: abcd) or Google drive and put it in the folder (-->results-->fashion).
Test by yourself
python test.py --name=fashion --model=painet --gpu_ids=0
If you use this code, please cite our paper.
@inproceedings{PISE,
title={{PISE}: Person Image Synthesis and Editing with Decoupled GAN},
author={Jinsong, Zhang and Kun, Li and Yu-Kun, Lai and Jingyu, Yang},
booktitle={Computer Vision and Pattern Recognition (CVPR)},
year={2021}
}
Our code is based on GFLA.