tf.app.flags.FLAGS函数的使用

最新推荐文章于 2021-08-20 21:39:41 发布

qq_36963963

最新推荐文章于 2021-08-20 21:39:41 发布

阅读量468

点赞数

分类专栏：深度学习函数汇总文章标签：深度学习 tensorflow

本文链接：https://blog.csdn.net/qq_36963963/article/details/101596355

版权

本文介绍了TensorFlow中tf.app.flags模块的使用，它类似于Python的argparse，用于定义并接收命令行参数。文章通过示例解释了DEFINE_string、DEFINE_boolean和DEFINE_integer等函数的用法。

摘要由CSDN通过智能技术生成

在源代码中大量使用了tf.app.flags，Tensorflow中的tf.app.flags与argparse模块有点类似，通过它们都可以定义输入参数，用于支持接受命令行传递参数，相当于接受argv。

tf.app.flags有tf.app.flags.DEFINE_string、tf.app.flags.DEFINE_boolean、tf.app.flags.DEFINE_integer等几类。

下面举个列子来说明怎么使用tf.app.flags。

import tensorflow as tf
 
# 第一个是参数名称，第二个参数是默认值(这里是‘model_dir’的路径)，第三个是参数描述,如果不想描述可以直接用 ''
tf.app.flags.DEFINE_string('model_dir', 'AgeGenderDeepLearning-master/Folds/tf/test_fold_is_0/run-7656/',
                           "Model directory (where training data lives)")
tf.app.flags.DEFINE_integer('max_steps', 20000, "description")
tf.app.flags.DEFINE_boolean('single_look', False, "single look at the image or multiple crops")
 
FLAGS = tf.app.flags.FLAGS
 
def main(_):
    print(FLAGS.model_dir)
    print(FLAGS.max_steps)
    print(FLAGS.single_look)
 
if __name__ == '__main__':
    tf.app.run()
<