“dest_directory = FLAGS.model_dir”是什么意思？

Question

我试图了解TenserFlow图像分类。从GitHub跟随code，从“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'。