simple-HRNet

收录理由

HRNet 的官方研究代码功能强大，但上手门槛不低，这个项目正是为此而生。它把高分辨率网络封装进一个简洁的 SimpleHRNet 类，能直接加载官方预训练权重，对单张或整批图片输出关节点预测，并搭配 YOLOv3、YOLOv3-tiny 或 YOLOv5 实现多人检测。仓库里备好了可直接运行的脚本，涵盖摄像头实时演示、从视频中提取关键点存为 CSV、用 TensorRT 加速推理，以及在 COCO 上训练模型；配套的 Colab 笔记本还能带你完整走一遍流程。代码同时支持 Windows 和 Linux，推理结果与原版实现一致，很适合作为原型开发的底座，比如做健身动作计数或工位人体工学分析。不过要把它当作起点而非长期维护的生产库，实际部署时还需自行调整。