Question

我是TensorFlow的新手。我按照以下示例尝试微调Inception v3模型： https://github.com/tensorflow/models/tree/master/slim#Tuning

我已经运行了脚本：

python train_image_classifier.py \
--train_dir=${TRAIN_DIR} \
--dataset_name=flowers \
--dataset_split_name=train \
--dataset_dir=${DATASET_DIR} \
--model_name=inception_v3 \
--checkpoint_path=${PRETRAINED_CHECKPOINT_DIR}/inception_v3.ckpt \
--checkpoint_exclude_scopes=InceptionV3/Logits,InceptionV3/AuxLogits \
--trainable_scopes=InceptionV3/Logits,InceptionV3/AuxLogits \
--learning_rate=0.01 \
--learning_rate_decay_type=fixed \
--save_interval_secs=60 \
--save_summaries_secs=60 \
--log_every_n_steps=100 \
--weight_decay=0.00004

生成model.ckpt-XXXX文件。如何将此文件转换为oder中的pb文件，以便在label_image示例（https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/label_image）中使用它？

感谢。

Answer 1

model.ckpt文件只有变量检查点，但没有graph def。您是否注意到有另一个名为model-ckpt-xxxxx.meta的文件？该文件有graph def。

from tensorflow.python.framework import meta_graph
from tensorflow.python.training import training_util

mg = meta_graph.read_meta_graph_file("model-ckpt-xxxxx.meta")
training_util.write_graph(mg.graph_def, "your/output/dir", "graph.pb", as_text=False)

TenssorFlow：微调Inception v3进行图像分类

1 个答案: