android - Tensorflow:构建图形和标签文件形成检查点文件

Question

我想从 inception-resnet-v2.ckpt 文件构建图形和标签文件。我已经下载了检查点文件表格wget http://download.tensorflow.org/models/inception_resnet_v2_2016_08_30.tar.gz。

我想用 inception-resnet-v2 替换 tensorflow: android camera domo 应用程序中的 inception5h 模型。这需要一个 MODEL_FILE 和一个 LABEL_FILE 。

现在我不知道如何从检查点文件中获取 .pb 文件和标签文件。我正在学习 tensorflow，仍处于初级水平。

Answer 1

不确定标签文件是什么，但要将检查点转换为 .pb 文件(二进制 protobuf)，您必须 freeze the graph .这是我使用的脚本:

#!/bin/bash -x

# The script combines graph definition and trained weights into
# a single binary protobuf with constant holders for the weights.
# The resulting graph is suitable for the processing with other tools.


TF_HOME=~/tensorflow/

if [ $# -lt 4 ]; then
    echo "Usage: $0 graph_def snapshot output_nodes output.pb"
    exit 0
fi

proto=$1
snapshot=$2
out_nodes=$3
out=$4

$TF_HOME/bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=$proto \
    --input_checkpoint=$snapshot \
    --output_graph=$out \
    --output_node_names=$out_nodes 

这里proto是一个Graph定义(text protobuf)，snapshot是一个checkpoint。