Windows上的Tensorflow Retrain

时间:2017-01-02 21:07:38

标签: python windows machine-learning tensorflow

当我按照" How to Retrain Inception's Final Layer for New Categories"的教程时,我需要像这样构建保留器

bazel build tensorflow/examples/image_retraining:retrain

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

提前谢谢

3 个答案:

答案 0 :(得分:6)

在我的情况下,tensorflow版本是1.2,相应的retrain.py是hereenter image description herehere下载并提取鲜花图像。 现在将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 branch并下载label_image.py代码,如下所示
  • enter image description here
  • 然后运行python label_image.py --image=image/path/to/test/classfication --graph=retrained_graph.pb --labels=retrained_labels.txt
  • 结果将类似于enter image description here

答案 1 :(得分:4)

从屏幕截图中看来,您已经安装了TensorFlow PIP软件包,而image retraining tutorial中的说明假设您克隆了Git repository(并且可以使用bazel来构建TensorFlow)。

但是,幸运的是,用于图像再训练的脚本(retrain.py)是一个简单的Python脚本,您可以在不构建任何内容的情况下下载和运行该脚本。只需从TensorFlow存储库的分支下载retrain.py的副本(例如,如果您已经安装了TensorFlow 0.12,就可以下载this version),您应该能够通过在命令提示符下键入python retrain.py来运行它。

答案 2 :(得分:1)

我在Windows上遇到了同样的问题。我的窗户找不到script.retrain。我从here的tensoflow网站下载了retrain.py文件。然后,将文件复制到tensorflow文件夹中,并使用Python命令运行重新训练脚本。