昇腾社区首页
中文
注册

准备TensorFlow模型npy数据文件

前提条件

准备好pb模型(本样例采用的ResNet-50)。

操作步骤

  1. 进入pb文件所在目录。
  2. 创建dump脚本文件tools_tensorflow_dump.py(代码参考)。
  3. 执行如下命令dump数据。
    指定pb文件、模型推理输入bin文件、输入节点名称等。
    python3 tools_tensorflow_dump.py resnet50_tensorflow_1.7.pb -i test.bin -n 'Placeholder:0' -s [1,224,224,3] -a
  4. 生成结果。

    生成的npy文件如下图所示。