【天天向上】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 --scale_values=data --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相关内容。(请高手赐教)
页:
[1]