Skip to content

rongtuech/handbodypose

Repository files navigation

Hand and body pose


1) Introduce:

This Git is the combination of lightweight openpose model (training/testing code) + mediapipe library to extract the full body pose and hand pose. The inference is also included optical flow algorithm for speeding up.

The light weight openpose model and post-process code is modified from Daniil-Osokin repository. Please check his original git for reference.


2) Requirements:

  • Python 3.5 (or above)
  • CMake 3.10 (or above)
  • C++ Compiler (g++ or MSVC)
  • OpenCV 4.0 (or above)
  • MediaPipe

3) Usage:

Repository includes:

  • Dataloader from COCO detection dataset.
  • Training code.
  • ONNX export code.
  • c++/python postprocess (from Daniil-Osokin git).
  • Inference code (combine optical flow and lightweight openpose) -> get hand segment by body pose -> process via MediaPipe.

4) Future work:

  • Combine with MediaPipe lib (Done).
  • Improve Optical flow process.
  • ONNX/TensorRT python inference code.
  • c++ inference.

5) Current result:

Model FPS-PYTORCH FPS-ONNX FPS-TensorRT
+ Optical flow algorithm - Hand joints detection ~ 50 x x
- Optical flow algorithm - Hand joints detection ~ 26 x x
+ Optical flow algorithm + Hand joints detection ~ 22 ~ 29 ~ 34
- Optical flow algorithm + Hand joints detection ~ 14 ~ 22 ~ 30

Pose 10 Pose 2 Pose 3 Pose 4 Pose 5

PLEASE CHECK THE ATTACHED LICENSE FOR USING

Releases

No releases published

Packages

No packages published