convolutional-pose-machines-tensorflow

收录理由

这个项目把卷积姿态机（Convolutional Pose Machines）用 TensorFlow 重新实现了一遍，专门做二维人体和手部关键点估计。它没有把代码塞进某个庞大的框架里，而是保持了独立可运行的状态，自带身体和手部姿态的演示脚本，既支持图片和视频文件，也能直接调用摄像头做实时演示。实现里还额外加了卡尔曼滤波做平滑，以及一个简单的手部跟踪模块，多阶段的图结构搭起来也比较直观，适合想研究 staged heatmap 逐步精修思路的人，或者打算用自己的关键点标注数据重新训练的情况。需要留意的是，它基于 TensorFlow 1.4 编写，而且模型默认目标大致位于画面中央，实际场景中通常得在前面接一个检测器来框出目标。项目里还提供了一种蒸馏训练方案，可以把较大的教师网络压缩成较小的学生模型。