💃 Mobile 2D Single Person (Or Your Own Object) Pose Estimation for TensorFlow 2.0

This repository is forked from edvardHua/PoseEstimationForMobile when the original repository was closed.
edvardHua/PoseEstimationForMobile repository is reopened! I'll maintain it separately. 👍

This repository currently implemented the Hourglass model using TensorFlow 2.0 with Keras API.

Table of contents

• Goals
• Getting Started
• Results
• Converting To Mobile Model
• Tuning
• Details
  • Folder Structure
  • Main Components
• TODO
• Related Projects
• Acknowledgements
• Reference
• Contributing
• License

Goals

• 📚 Easy to train
• 🏃‍ Easy to use the model on mobile device

Getting Started

Install Anaconda (~10 min)

• How To Install Anaconda on Ubuntu 18.04 [Quickstart]
• How to Install Anaconda on CentOS 7

Create Virtual Environment (~2 min)

Create new environment.

conda create -n {env_name} python={python_version} anaconda
# in my case
# conda create -n mpe-env-tf2-alpha0 python=3.7 anaconda

Start the environment.

source activate {env_name}
# in my case
# source activate mpe-env-tf2-alpha0

Install the requirements (~1 min)

cd {tf2-mobile-pose-estimation_path}
pip install -r requirements.txt
pip install git+https://github.com/philferriere/=PythonAPI

python downloader.py --download-path=COCO_DATASET_PATH

Run The Project

In order to use the project you have to:

1. Prepare the dataset(ai_challenger dataset) and unzip.
2. Run the model using:

python train.py \
--dataset_config config/dataset/coco_single_person_only-gpu.cfg \
--experiment_config config/training/coco_single_experiment01-cpm-sg4-gpu.cfg

Compatiable Datasets

• ai challenge's keypoint names: ['top_head', 'neck', 'left_shoulder', 'right_shoulder', 'left_elbow', 'right_elbow', 'left_wrist', 'right_wrist', 'left_hip', 'right_hip', 'left_knee', 'right_knee', 'left_ankle', 'right_ankle']
• coco's keypoint names: ['nose', 'left_eye', 'right_eye', 'left_ear', 'right_ear', 'left_shoulder', 'right_shoulder', 'left_elbow', 'right_elbow', 'left_wrist', 'right_wrist', 'left_hip', 'right_hip', 'left_knee', 'right_knee', 'left_ankle', 'right_ankle']

Results

AI Challenge Dataset

COCO Single persononly Dataset

Converting To Mobile Model

TensorFLow Lite

If you train the model, it will create tflite models per evaluation step.

Core ML

Check convert_to_coreml.py script. The converted .mlmodel support iOS14+.

Details

This section will be separated to other .md file.

Folder Structure

tf2-mobile-pose-estimation
├── config
|   ├── model_config.py
|   └── train_config.py
├── data_loader
|   ├── data_loader.py
|   ├── dataset_augment.py
|   ├── dataset_prepare.py
|   └── pose_image_processor.py
├── models
|   ├── common.py
|   ├── mobilenet.py
|   ├── mobilenetv2.py
|   ├── mobilenetv3.py
|   ├── resnet.py
|   ├── resneta.py
|   ├── resnetd.py
|   ├── senet.py
|   ├── simplepose_coco.py
|   └── simpleposemobile_coco.py
├── train.py            - the main training script
├── common.py 
├── requirements.txt
└── outputs             - this folder will be generated automatically when start training
    ├── 20200312-sp-ai_challenger
    |   ├── saved_model
    |   └── image_results
    └── 20200312-sp-ai_challenger
        └── ...

My SSD    
└── datasets            - this folder contains the datasets of the project.
    └── ai_challenger
        ├── train.json
        ├── valid.json
        ├── train
        └── valid

TODO

• Save model to saved_model
• Convert the model(saved_model) to TFLite model(.tflite)
• Convert the model(saved_model) to Core ML model(.mlmodel)
• Run the model on iOS
• Release 1.0 models
• Support distributed GPUs training
• Make DEMO gif running on mobile device
• Run the model on Android

Reference

[1] Paper of Convolutional Pose Machines
[2] Paper of Stack Hourglass
[3] Paper of MobileNet V2
[4] Repository PoseEstimation-CoreML
[5] Repository of tf-pose-estimation
[6] Devlope guide of TensorFlow Lite
[7] Mace documentation

Related Projects

• tucan9389/PoseEstimation-CoreML
• tucan9389/PoseEstimation-TFLiteSwift (Preparing...)
• tucan9389/KeypointAnnotation
• osmr/imgclsmob
• edvardHua/PoseEstimationForMobile
• jwkanggist/tf-tiny-pose-estimation
• dongseokYang/Body-Pose-Estimation-Android-gpu

Other Pose Estimation Projects

• cbsudux/awesome-human-pose-estimation

Contributing

This section will be separated to other .md file.

Any contributions are welcome including improving the project.

License

Apache License 2.0