python - "dest_directory = FLAGS.model_dir"是什么意思？

Question

我正在尝试理解 TenserFlow 图像分类。已关注 code来自 GitHub，从“retrain.py”脚本中的 298 行开始。

dest_directory = FLAGS.model_dir
if not os.path.exists(dest_directory):
os.makedirs(dest_directory)

“FLAGS.model_dir” 是什么意思以及该目录位于何处？

Answer 1

FLAGS 保存已解析的命令行参数。该脚本使用 argparse库，但样式继承自 gflags库，最初在 Google 内部使用 C++ 开发，然后开源并移植到不同的语言。

从解析器定义中很容易看出FLAGS.model_dir的含义:

parser.add_argument(
      '--model_dir',
      type=str,
      default='/tmp/imagenet',
      help="""\
      Path to classify_image_graph_def.pb,
      imagenet_synset_to_human_label_map.txt, and
      imagenet_2012_challenge_label_map_proto.pbtxt.\
      """
  )

因此，它的位置是由用户运行脚本时指定的。如果未指定任何内容，则使用此路径:'/tmp/imagenet'。