如何使用tensorflow的retrain.py重新训练多个初始实例

Posted 2021-04-06

我正在构建一个允许用户通过GUI重新启动的应用程序。我目前基本上正在通过tensorflow的image_retrain示例找到here并构建一个GUI作为前端。我不知道实际训练的神经网络存储在哪里或如何训练它的多个实例。似乎output_graph参数默认为/tmp/output_graph.pb，但我不知道这是否是经过训练的网络本身或其他东西（我不太了解tensorflow）。我还希望所有内容都只存储在我的应用程序运行的目录中。我目前还在克隆整个tensorflow github存储库，我想知道我需要的是retrain.py文件和label_image.py文件。这是我的问题： 1.什么是输出图？ 2.如何存储多个重新训练的初始实例，每个实例都针对特定的类别集进行培训？ 3.我是否需要克隆整个存储库？ 我在Windows上运行python 3.5.2 64位和tensorflow版本1.5.0

答案

1. 输出图是网络本身的存储位置
2. 为您的初始实例创建一个目录。在主目录中创建一个瓶颈目录。运行--output_graph /path/to/main/dir/output_graph.pb --bottleneck_dir /path/to/main/dir/bottleneck_dir --output_labels /path/to/main/dir/output_labels.txt时使用额外的参数retrain.py。 然后，当你运行label_image.py时，你应该有另外的参数，如下所示：--graph /path/to/main/dir/output_graph.pb --labels=/path/to/main/dir/output_labels.txt和你将根据tensorflow tutorial使用的其他参数
3. 不，你只需要retrain.py和label_image.py