Simultaneous human pose estimation and segmentation

Inference Python script for simultaneous human pose estimation and segmentation MXNet models.

Installation

pip3 install -r requirements.txt

Download model files form the repo releases.

Usage

python3 demo_mx_hpeseg.py --in-video=<path_to_your_video> --out-video=<path_to_resulting_video> --model=<path_to_model_stem> 
python3 demo_tlite_hpeseg.py --in-video=<path_to_your_video> --out-video=<path_to_resulting_video> --model=<path_to_tfl_model>
python3 demo_onnx_hpeseg.py --in-video=<path_to_your_video> --out-video=<path_to_resulting_video> --model=<path_to_onnx_model>

Example:

python3 demo_mx_hpeseg.py --in-video=../hpeseginf_data/vid2.mp4 --out-video=../hpeseginf_data/hpeseg_result.mp4 --model=../hpeseginf_data/ducnet_mobilenet_w1_coco