13 Star 61 Fork 279

Ascend/ModelZoo-TensorFlow

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

中文|English

AlignedReID Tensorflow离线推理

此链接提供AlignedReID 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/AlignedReID_for_ACL

2. 下载数据集和预处理

  1. 请自行下载cuhk03数据集

  2. 将整个数据集放至 scripts/data

3. 离线推理

离线模型转换

  • 环境变量设置

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

  • Pb模型转换为om模型

    pb模型下载链接

    atc --model=AlignedReID_tf.pb --framework=3 --output=AlignedReID_100batch --output_type=FP32 --soc_version=Ascend310 --input_shape="images_total:100,224,224,3" --log=info
    
  • 编译程序

    bash build.sh
    
  • 开始运行:

    cd scripts
    bash benchmark_tf.sh
    

性能

结果

本结果是通过运行上面适配的推理脚本获得的。要获得相同的结果,请按照《快速指南》中的步骤操作。

推理精度结果

model data Mean-cmc
offline Inference cuhk03 val 99.4%

参考

[1] https://github.com/Phoebe-star/AlignedReID

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

搜索帮助