Home

Awesome

update 2019-08-14

Add tensorflow 2.0 version, link here. Repo still need to update. Welcome to raise questions.

update 2019-04-01

update 2019-03-20

Add evaluation part. The evaluation based on ai-challenger evaluation solution. More information please refer to this link.

Easy way for how to use it:

update 2019-03-18

Based on official pytorch implementation, re-write tf-model, see lightweight_openpose.py for detailed information, corresponding net structure picture is named lightweight.png. New pre_trained model will be upload soon.

A tensorflow implementation about Arxiv Paper "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose"

Thanks for the author provide PyTorch version

(This repo is more useful than mine i think, enjoy it!) Pytorch implementation.

trained model

update

The original caffe prototxt(provided by paper author) has been upload, you can found in repo file named "lightweight_openpose.prototxt"

Requirement

Train

python3 train.py

all parameters has been set in src/train_config.py.

Train Dataset

we use ai-challenger format dataset, which can found in this website.

Note