有没有办法在TensorFlow中修改inception-v3的python代码?

时间:2016-07-12 17:11:32

标签: python machine-learning neural-network tensorflow

现在我正在使用基于谷歌提供的教程的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脚本运行?

2 个答案:

答案 0 :(得分:2)

不幸的是,该模型是用C ++编写的。

但是tensorflow/models repo中有一个非常好的初始实现:https://github.com/tensorflow/models/tree/master/inception

它还包括有关如何训练和评估模型的详细说明。 评估步骤描述于 https://github.com/tensorflow/models/tree/master/inception#how-to-evaluate 并用Python编写。 评估脚本为inception_eval.py

答案 1 :(得分:1)

用于模型的python脚本(用tensorflow keras编写)

  • VGG16
  • VGG19
  • ResNet50
  • Inception v3
  • 用于音乐标记的CRNN

可以找到here