#278 · Primary category: Computer Vision

lightweight-human-pose-estimation.pytorch

coco-keypoints-detection deep-learning human-pose-estimation lightweight lightweight-openpose mscoco-keypoint openpose openvino pose-estimation pytorch real-time

Fast and accurate human pose estimation in PyTorch. Contains implementation of "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose" paper.

Project last updated:04/30/24

GitHub Stars

2.2K

Forks

495

Contributors

1

License

Apache-2.0

Why we included this project

This repo implements the Lightweight OpenPose paper, which compresses the original OpenPose approach so that multi-person pose estimation runs in real time on a plain CPU instead of needing a GPU. It detects up to 18 body keypoints per person, covering the full skeleton from head to ankles, with only a small accuracy trade-off versus the heavier model. The repository includes training code for COCO 2017, validation scripts, and working Python and C++ demos, plus a pre-trained model so you can test it right away. That makes it a useful starting point for engineers shipping pose detection on edge devices, and for anyone who wants to see how a complex keypoint-and-connection network can be slimmed down without collapsing.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category