mnist的Tensorflow官方模板（fullly_connected_feed.py文件中参数解析问题）-优快云博客

本文链接：https://blog.youkuaiyun.com/dengchunlin01/article/details/79108399

本文介绍TensorFlow中使用argparse设置输入数据目录和日志目录的方法。通过os.path.join和os.getenv组合使用，灵活指定默认路径。同时解释了如何自动下载并读取数据集。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

  parser.add_argument(
      '--input_data_dir',
      type=str,
      default=os.path.join(os.getenv('TEST_TMPDIR', '/tmp'),
                           'tensorflow/mnist/input_data'),
      help='Directory to put the input data.'
  )
  parser.add_argument(
      '--log_dir',
      type=str,
      default=os.path.join(os.getenv('TEST_TMPDIR', '/tmp'),
                           'tensorflow/mnist/logs/fully_connected_feed'),
      help='Directory to put the log data.'
  )

该部分代码主要用来对--input_data_dir（输入数据路径）和--log_dir（日志数据路径）进行解析。

疑惑代码：os.path.join(os.getenv('TEST_TMPDIR', '/tmp'),'tensorflow/mnist/input_data')

理解：

os.path.join(a, b)用于将路径a和b进行链接，即a/b。

os.getenv(env_variable_name)用于获取环境变量的值，默认为None。

os.getenv('TEST_TMPDIR')返回值为None。（Linux和Windows均有测试）

因此这里可以将default直接赋值为相关路径即可。

但至于模块为什么要写成这样就不太了解了，主要是不知道TEST_TMPDIR是什么环境变量。

input_data.read_data_sets(path)会自动去路径中寻找输入数据文件，如果输入数据不存在，则会自动进行下载。（但下载是从外网下载，所以需要翻墙）