基于keras的实时多人姿态识别估计
依赖：
Keras
Caffe(docker)
测试：
1，把keras模型model.h5下载下来。（注意此文件在dropbox上，国内下载不了）
2，运行demo.ipynb.
3，运行图像demo，python demo_image.py --image sample_images/ski.jpg
训练：
1.安装gsuntil
2.下载数据集(约25G)，cd dataset; sh get_dataset.sh
3.下载COCO toolbox，cd coco/PythonAPI; sudo python setup.py install
4.进入训练文件夹 cd ../../../training
5.运行训练 python train_pose.py
因为目前keras模型此文件在dropbox上，国内下载不了，故要自己生成h5模型，
这样就需要把caffe模型转换为keras模型。
这时要先安装caffe docker。此步骤网上都有，直接参考即可。
之后，按如下步骤进行转换：
1.下载caffe模型，sh get_caffe_model.sh
2. dump numpy data，docker run -v [absolute path]:/workspace -it bvlc/caffe:cpu python dump_caffe_layers.py
3. 运行转换脚本，python caffe_to_keras.py
分享到:
閱讀更多 AI踐行者 的文章
