当我按照" How to Retrain Inception's Final Layer for New Categories"的教程时,我需要像这样构建保留器
bazel build tensorflow/examples/image_retraining:retrain
但是,我在Windows上的张量流没有这样的目录。我想知道为什么以及如何解决这个问题?
提前谢谢
答案 0 :(得分:6)
在我的情况下,tensorflow版本是1.2,相应的retrain.py是here。 从here下载并提取鲜花图像。 现在将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
注意:上述命令中的最后两个参数是可选的。
现在测试重新训练的模型:
python label_image.py --image=image/path/to/test/classfication --graph=retrained_graph.pb --labels=retrained_labels.txt
答案 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命令运行重新训练脚本。