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

最新推荐文章于 2023-10-20 09:15:23 发布

tan197

最新推荐文章于 2023-10-20 09:15:23 发布

阅读量6.2k

点赞数

本文链接：https://blog.youkuaiyun.com/tan197/article/details/84502097

版权

本文介绍如何使用TensorFlow定义和配置参数，包括设置训练图片所在文件夹及batch大小，并展示了如何利用这些参数读取指定文件夹下的所有jpg图片文件路径。

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

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文件路径