13 Star 62 Fork 280

Ascend/ModelZoo-TensorFlow

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

中文|English

OpenPose TensorFlow离线推理

此链接提供OpenPose TensorFlow模型在NPU上离线推理的脚本和方法

注意

此案例仅为您学习Ascend软件栈提供参考,不用于商业目的。

在开始之前,请注意以下适配条件。如果不匹配,可能导致运行失败。

Conditions Need
CANN版本 >=5.0.3
芯片平台 Ascend310/Ascend310P3
第三方依赖 请参考 'requirements.txt'

快速指南

1. 拷贝代码

git clone https://gitee.com/ascend/ModelZoo-TensorFlow.git
cd Modelzoo-TensorFlow/ACL_TensorFlow/built-in/cv/OpenPose_for_ACL

2. 下载数据集和预处理

请自行下载COCO2014测试数据集, 详情见: dataset

3. 获取pb模型

获取OpenPose pb模型, 详情见: models

4. 获取处理脚本

pafprocess、slidingwindow 下载链接: tf_openpose and put them into libs

5. 离线推理

数据预处理

python3 preprocess.py \
    --resize 656x368 \
    --model cmu \
    --coco-year 2014 \
    --coco-dir ../dataset/coco/ \
    --output-dir ../input/

Pb模型转换为om模型

  • 环境变量设置

    请参考说明,设置环境变量

  • Pb模型转换为om模型

    pb模型下载链接

    atc --framework=3 \
        --model=./models/OpenPose_for_TensorFlow_BatchSize_1.pb \
        --output=./models/OpenPose_for_TensorFlow_BatchSize_1 \
        --soc_version=Ascend310 \
        --input_shape="image:1,368,656,3"
    

编译程序 编译推理应用程序, 详情见: xacl_fmk

开始运行

/xacl_fmk -m ./models/OpenPose_for_TensorFlow_BatchSize_1.om \
    -o ./output/openpose \
    -i ./input \
    -b 1

后处理

python3 postprocess.py \
    --resize 656x368 \
    --resize-out-ratio 8.0 \
    --model cmu \
    --coco-year 2014 \
    --coco-dir ../dataset/coco/ \
    --data-idx 100 \
    --output-dir ../output/openpose 

样本脚本 我们还支持使用predict_openpose.sh运行上述所有步骤,构建程序除外

6.结果


OpenPose 推理 :

Type IoU Area MaxDets Result
Average Precision (AP) 0.50:0.95 all 20 0.399
Average Precision (AP) 0.50 all 20 0.648
Average Precision (AP) 0.75 all 20 0.400
Average Precision (AP) 0.50:0.95 medium 20 0.364
Average Precision (AP) 0.50:0.95 large 20 0.443
Average Recall (AR) 0.50:0.95 all 20 0.456
Average Recall (AR) 0.50 all 20 0.683
Average Recall (AR) 0.75 all 20 0.465
Average Recall (AR) 0.50:0.95 medium 20 0.371
Average Recall (AR) 0.50:0.95 large 20 0.547

参考

[1] https://github.com/ildoonet/tf-pose-estimation/tree/master/tf_pose/pafprocess/

马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/ascend/ModelZoo-TensorFlow.git
git@gitee.com:ascend/ModelZoo-TensorFlow.git
ascend
ModelZoo-TensorFlow
ModelZoo-TensorFlow
master

搜索帮助