Tensorflow中tf.flags的作用及使用方法

import tensorflow as tf
FLAGS = tf.flags.FLAGS
tf.flags.DEFINE_string('buckets', 'oss://{OSS Bucket}/', '训练图片所在文件夹')  
tf.flags.DEFINE_string('batch_size', '15', 'batch大小')
files = tf.gfile.Glob(os.path.join(FLAGS.buckets,'*.jpg')) # 如我想列出buckets下所有jpg文件路径

在老版本的tensorflow中使用tf.app.flags来定义参数,新版本用tf.flags来定义参数。

你可能感兴趣的:(Tensorflow中tf.flags的作用及使用方法)