Android Tensorflow：从检查点文件构建图形和标签文件

我想从inception-resnet-v2.ckpt文件构建图形和标签文件。我已经下载了检查点文件表单

wgethttp://download.tensorflow.org/models/inception_resnet_v2_2016_08_30.tar.gz

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

模型文件

和一个

标签文件

现在我不知道如何从检查点文件中获取.pb文件和标签文件。

---

我正在学习tensorflow，仍处于初级阶段。

不确定标签文件是什么，但要将检查点转换为.pb文件（即二进制protobuf），您必须。下面是我使用的脚本：

#!/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是一个图形定义（text protobuf），snapshot是一个检查点

冻结模型后，需要对其进行优化

看看这个，太棒了

对于您可以获得的标签（贷记到）

---

感谢您回复@Dmytro。但这对我没有帮助。

bazel build tensorflow/python/tools:optimize_for_inference

bazel-bin/tensorflow/python/tools/optimize_for_inference \
--input=/tf_files/retrained_graph.pb \
--output=/tf_files/optimized_graph.pb \
--input_names=Mul \
--output_names=final_result