查看: 88|回复: 0

[平台测评] 【天天向上】OpenVINO学习笔记(九)3d人体姿态估计测试

[复制链接]
本帖最后由 云天 于 2020-6-24 18:34 编辑

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
没有发现匹配的模型
16.png
2、运行前要先安装pytorch,否则会出如下错误:
10.png
如果没有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时使用了管理员权限。
smart_classroom4.png

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


15.png

(3)安装中需要确认:Y

17.png

4、演示成功图片
python demo.py --model human-pose-estimation-3d.pth --video 0 --device CPU

14.png

很慢,我的机器配置也比较差,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
出现如下错误:
13.png

解决方法:
将models及moduels文件夹复制到scripts文件夹中。
12.png

(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
转换时发生如下错误:

21.png


(4)暂未能进行如下测试:

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


(5)使用OpenVINO自带的human_pose_estimation_3d_demo.py出现如下错误:
22.png

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

pose_extractor文件夹内容如下:

23.png

没有extract_poses相关内容。(请高手赐教)

高级模式
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

为本项目制作心愿单
购买心愿单
心愿单 编辑
[[wsData.name]]

硬件清单

  • [[d.name]]
btnicon
我也要做!
点击进入购买页面
上海智位机器人股份有限公司 沪ICP备09038501号-4

© 2013-2020 Comsenz Inc. Powered by Discuz! X3.4 Licensed

mail