Implementation of "Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields" for AI challenger keypoint competition

Train demo

install cython package

./cython/rebuild.sh

Generate intermediate files

change folder name and json name in pose_io/parse_label.py

path1 = '/data/guest_users/liangdong/liangdong/practice_demo/AIchallenger/keypoint_validation_annotations_20170911.json' 
trainimagepath = '/data/guest_users/liangdong/liangdong/practice_demo/AIchallenger/validation_image/keypoint_validation_images_20170911/'

python pose_io/parse_label.py

Train

python TrainWeight.py

You can download mxnet model and parameters for vgg19 from here

Cite paper Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields

@article{cao2016realtime,
  title={Realtime Multi-Person 2D Pose Estimation using Part Affinity Fields},
  author={Zhe Cao and Tomas Simon and Shih-En Wei and Yaser Sheikh},
  journal={arXiv preprint arXiv:1611.08050},
  year={2016}
  }