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}
}