目前我正在使用Google提供的教程中的Inception-v3模型。为了重新训练和测试,我使用以下命令:
bazel build tensorflow/examples/label_image:label_image && \bazel-bin/tensorflow/examples/label_image/label_image \--graph=/tmp/output_graph.pb --labels=/tmp/output_labels.txt \--output_layer=final_result \--image=$HOME/flower_photos/daisy/21652746_cc379e0eea_m.jpg
但现在我需要修改代码的某些部分。由于我使用Bazel进行构建,我不太确定这是否可能,也不知道代码具体在哪里。是否有办法将其作为普通的Python代码运行,像常规的Python脚本那样执行?
回答:
遗憾的是,该模型是用C++编写的。
但是在tensorflow/models
仓库中有一个非常好的Inception实现,链接在这里:https://github.com/tensorflow/models/tree/master/inception
它还包括了如何训练和评估模型的详细说明。评估步骤在https://github.com/tensorflow/models/tree/master/inception#how-to-evaluate中有描述,并且是用Python编写的。评估脚本是inception_eval.py
。