Skip to content

person_transfer

Directory actions

More options

Directory actions

More options

Latest commit

 

History

History

person_transfer

README.md

Person Image Generation

Code for person image generation. This is Pytorch implementation for pose transfer on both Market1501 and DeepFashion dataset.

Requirement

  • pytorch 1.0.1
  • torchvision
  • dominate
  • Others

Getting Started

Installation

  • Clone this repo:
git clone https://github.com/Ha0Tang/SelectionGAN
cd SelectionGAN
cd person_transfer

Data Preperation

We use OpenPose to generate keypoints. We also provide the prepared images for convience.

Market1501

sh datasets/download_selectiongan_dataset.sh market_data

DeepFashion

sh datasets/download_selectiongan_dataset.sh fashion_data

Training

Market-1501

sh train_market.sh

DeepFashion

sh train_fashion.sh

Testing

Market1501

sh test_market.sh

DeepFashion

sh test_fashion.sh

Pretrained Models

Market1501

sh scripts/download_selectiongan_model.sh market

DeepFashion

sh scripts/download_selectiongan_model.sh fashion

Evaluation

Follow Pose-Transfer for more details.