【天天向上】OpenVINO学习笔记（九）3d人体姿态估计测试

3d人体姿态估计测试

github链接：https://github.com/Daniil-Osokin/lightweight-human-pose-estimation-3d-demo.pytorch

（一）CPU版测试：

1、先下载模型

下载预训练模型：Google Drive

注：使用如下命令

python downloader.py --name human-pose-estimation-3d --output_dir C:\Users\zlzx\Documents\Intel

没有发现匹配的模型

2、运行前要先安装pytorch,否则会出如下错误：

如果没有GPU（仅用于CPU），请使用以下命令：

• 版本1.2： conda install pytorch==1.2.0 torchvision==0.4.0 cpuonly -c pytorch
• 对于新版本： conda install pytorch torchvision cpuonly -c pytorch
  
  

要么如果您有GPU（使用自己的CUDA版本），请使用以下命令：

• 版本1.2 ：conda install pytorch == 1.2.0 torchvision == 0.4.0 cudatoolkit = 10.0 -c pytorch
• 对于新版本 ：conda install pytorch torchvision cudatoolkit = 10.1 -c pytorch
• 最好使用anaconda命令提示符安装pytorch，如果可能，请使用python 3.6。 它更合适。
  
  

3、在安装pytorch时可能会出现EnvironmentNotWritableError: The current user does not have write permissions to the target environment.的情况。其主要原因是用户没有对anaconda3文件夹的读写权限，造成其原因可能是由于在安装anaconda时使用了管理员权限。

（1）最简单的方法就是撤销对这个文件夹权限限制，定位到anaconda3同级文件夹下打开终端执行如下代码即可：
sudo chmod 777 -R anaconda3
（2）Windows下，修改文件夹属性

（3）安装中需要确认：Y

4、演示成功图片

python demo.py --model human-pose-estimation-3d.pth --video 0 --device CPU

很慢，我的机器配置也比较差，0.1FPS。

（二）用openvino加速

（1）去安装路径中开启openvino
<OpenVINO_INSTALL_DIR>/bin/setupvars.bat
（2）将checkpoint转换为ONNX
python scripts/convert_to_onnx.py --checkpoint-path human-pose-estimation-3d.pth
出现如下错误：

解决方法：
将models及moduels文件夹复制到scripts文件夹中。

（3）转成openvino的格式python <OpenVINO_INSTALL_DIR>/deployment_tools/model_optimizer/mo.py --input_model human-pose-estimation-3d.onnx --input=data --mean_values=data[128.0,128.0,128.0] --scale_values=data[255.0,255.0,255.0] --output=features,heatmaps,pafs

转换时发生如下错误：

（4）暂未能进行如下测试：

python demo.py --model human-pose-estimation-3d.xml --video 0 --device CPU --use-openvino

（5）使用OpenVINO自带的human_pose_estimation_3d_demo.py出现如下错误：

C:\Program Files (x86)\IntelSWTools\openvino\deployment_tools\open_model_zoo\demos\python_demos\human_pose_estimation_3d_demo\pose_extractor

pose_extractor文件夹内容如下：

没有extract_poses相关内容。（请高手赐教）