#474 · Primary category: Computer Vision

simple-HRNet

coco-dataset computer-vision deep-learning hrnet human-pose-estimation keypoint-detection machine-learning pytorch yolov3

Multi-person Human Pose Estimation with HRNet in Pytorch

Project last updated:12/29/22

GitHub Stars

608

Forks

111

Contributors

1

License

GPL-3.0

Why we included this project

HRNet's official research code is powerful but not exactly friendly to pick up, and this project exists to fix that. It wraps the high-resolution network in a small SimpleHRNet class that loads the official pre-trained weights and returns joint predictions for a single image or a whole batch, then adds multi-person detection by pairing HRNet with YOLOv3, YOLOv3-tiny, or YOLOv5. Ready-to-run scripts cover a live webcam demo, pulling keypoints to CSV from recorded video, TensorRT conversion for faster inference, and training on COCO, and a Colab notebook walks through the whole flow. The code runs on Windows and Linux and produces results that match the original implementation, which makes it a comfortable base for prototyping pose-based tools like fitness tracking or ergonomics analysis. Just treat it as a starting point rather than a maintained production library, and plan to adapt it for your own deployment.

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