How to generate the pose map and pose mask of my own dataset?

charliememory / Pose-Guided-Person-Image-Generation

Tensorflow implementation of our NIPS 2017 paper "Pose Guided Person Image Generation"

MIT License

301 stars 82 forks source link

How to generate the pose map and pose mask of my own dataset? #40

Open mogeko2333 opened 4 years ago

mogeko2333 commented 4 years ago

According to your code, generating 18 keypoints depends on file——annotation - {type}. csv, so how to generate my own data set's xxxx.csv file?

charliememory commented 4 years ago

Hi, I am not sure which code piece do you mean. As to the pose keypoints, they are generated from OpenPose and the script "datasets/convert_RCV.py" helps to convert the keypoints in the format of (row, column, visibility). There is also a more recent PyTorch implementation https://github.com/budui/Human-Pose-Transfer . Hoe it helps.