Windows上的Tensorflow Retrain

Question

当我按照" 如何重新启动新类别的最终图层 "教程时,我需要构建这样的保留器

bazel build tensorflow/examples/image_retraining:retrain

但是,我在Windows上的张量流没有这样的目录.我想知道为什么以及如何解决这个问题？

先感谢您

Answer 1

在我的情况下,tensorflow版本是1.2,相应的retrain.py在这里. 从这里下载并提取鲜花图像.现在运行retrain.py文件

python retrain.py --image_dir=path\to\dir\where\flowers\images\where\extracted --output_lables=retrained_labels.txt --output_graph=retrained_graph.pb

注意:上面命令中的最后两个参数是可选的.

现在测试再训练的模型:

• 转到master分支并下载label_image.py代码,如下所示
• 
• 然后跑 python label_image.py --image=image/path/to/test/classfication --graph=retrained_graph.pb --labels=retrained_labels.txt
• 结果就像

Answer 2

从屏幕截图中可以看出，您已经安装了TensorFlow PIP软件包，而图像再培训教程中的说明假定您已经克隆了Git存储库（并可以bazel用来构建TensorFlow）。

但是，幸运的retrain.py是，用于图像再训练的脚本（）是一个简单的Python脚本，您可以在不构建任何内容的情况下下载并运行该脚本。只需retrain.py从TensorFlow存储库的分支中下载与您已安装的软件包相匹配的副本（例如，如果您已安装TensorFlow 0.12，则可以下载此版本），并且应该能够通过python retrain.py在命令提示符下键入来运行它。