基于tensorflow的人类姿势估计

利用MPII人类姿势数据集进行训练,这是一个内容十分丰富的数据集,专门针对人类姿态估计。目前只有TensorFlow的实现。将人类姿态估计的任务应用到真实的图片中,他们的方法既解决了动作识别,也能进行估计,与之前先检测人类动作在对此进行推测的技术有所区分。在实施过程中用到了基于CNN的探测器和整数线性规划法。

依赖:

python3

scipy

scikit-image

matplotlib

pyyaml

easydict

cython

munkres

注意为了运行训练或预测脚本,把TF_CUDNN_USE_AUTOTUNE设置为0

演示代码运行:

单人:

下载预训练模型

$ cd models/mpii

$ ./download_models.sh

$ cd -

运行demo

$ TF_CUDNN_USE_AUTOTUNE=0 python3 demo/singleperson.py

多人:

进行编译依赖

$ ./compile.sh

下载预训练模型

$ cd models/coco

$ ./download_models.sh

$ cd -

运行demo

TF_CUDNN_USE_AUTOTUNE=0 python3 demo/demo_multiperson.py

查看原文 >>
相关文章