基于tensorflow的人类姿势估计
基于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
查看原文 >>