Question

我正在使用鲜花数据集从头开始训练初始模型。使用tensorflow模型提供的脚本。培训的输出是这些文件：

checkpoint
events.out.tfevents.xxxxxx
model.ckpt-xxxx.data-00000-of-00001
model.ckpt-xxxx.index
model.ckpt-xxxx.meta

model.ckpt-xxxx.data-00000-of-00001
model.ckpt-xxxx.index
model.ckpt-xxxx.meta

这些是我得到的一些文件。有人有一个脚本来转换这些文件我可以用来分类我的图像吗？我如何用它来测试我自己的图像？

Answer 1

这是一个三步过程。

第1步：希望你已经有了tensorflow models目录，因为你已经从中训练过了。运行以下与您拥有的models目录相关的命令：

python models/research/slim/export_inference_graph.py --model_name=<MODEL_NAME> --output_file=<NAME_OF_PB_FILE_CREATED> --dataset_dir=<PATH_TO_TF_RECORDS_DIRECTORY>

例如：

python models/research/slim/export_inference_graph.py --model_name=inception_v3 --output_file=/home/user1/inception_v3_inf_graph.pb --dataset_dir=/home/user1/tfRecords

第2步：克隆tensorflow github存储库。（git clone https://github.com/tensorflow/tensorflow.git）。将以下命令w.r.t运行到克隆的tensorflow目录，如下所示：

python tensorflow/tensorflow/python/tools/freeze_graph.py --input_graph=<PATH_TO_PB_FILE_CREATED_IN_STAGE1> --input_checkpoint=<PATH_TO_CKPT_FILES_GENERATED_DURING_TRAINING> --input_binary=true --output_graph=<PATH_TO_SAVE_OUTPUT_FROZEN_GRAPH> --output_node_names=<OUTPUT_NODE_NAMES_OF_MODEL>

例如：

python tensorflow/tensorflow/python/tools/freeze_graph.py --input_graph=/home/user1/inception_v3_inf_graph.pb --input_checkpoint=/home/user1/model.ckpt-50000 --input_binary=true --output_graph=/home/user1/frozen_inception_v3.pb --output_node_names=InceptionV3/Predictions/Reshape_1

请注意示例中的数字50000。这表示迭代次数。如果您已经训练了10次迭代的模型，那么它将是10.此外，即使每个检查点有3种类型的文件（元数据，数据和索引），我们只提到第一部分。其余的将由脚本自动解析。

步骤3：将以下命令w.r.t运行到克隆的tensorflow目录，如下所示：

python tensorflow/tensorflow/examples/label_image/label_image.py --image=<PATH_TO_TEST_IMAGE_FILE> --input_layer=input --output_layer=<MODEL_OUTPUT_LAYER_NAME> --graph=<PATH_TO_FROZEN_GRAPH_CREATED_IN_STAGE2> --labels=<PATH_TO_LABELS_FILE> --input_mean=<MEAN> --input_std=<STD_DEVIATION>

例如：

python tensorflow/tensorflow/examples/label_image/label_image.py --image=/home/user1/test_img.jpg --input_layer=input --output_layer=InceptionV3/Predictions/Reshape_1 --graph=/home/user1/frozen_inception_v3.pb --labels=/home/user1/labels.txt --input_mean=0 --input_std=255

最后一个将给出test_img.jpg

的预测结果

如何将我的model.ckpt文件转换为单个.ckpt文件？如何测试我的模型以预测图像分数？

1 个答案: