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.
- https://github.com/Daniil-Osokin/lightweight-human-pose-estimation.pytorch
- https://github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch
- Python 3.5 (or above)
- CMake 3.10 (or above)
- C++ Compiler (g++ or MSVC)
- OpenCV 4.0 (or above)
- MediaPipe
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.
- Combine with MediaPipe lib (Done).
- Improve Optical flow process.
- ONNX/TensorRT python inference code.
- c++ inference.
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 |
PLEASE CHECK THE ATTACHED LICENSE FOR USING