代码之家 › 专栏 › 技术社区 › user288609

关于tf.app.flags.define枚举的用法

tensorflow

1

user288609 · 技术社区 · 7 年前

在一个程序中,我看到了下面的代码段

tf.app.flags.DEFINE_enum('image_format', 'png', ['jpg', 'jpeg', 'png'],
                         'Image format.')

但我找不到 tf.app.flags.DEFINE_enum ,它的用途是什么?

1 回复 | 直到 7 年前

1

4

wicke 7 年前

这将生成一个命令行参数 image_format ,默认为 png ,哪个可以接受 PNG , jpeg 或 jpg . 在程序帮助字符串中,该标志将被解释为简单的“图像格式”。

从docstring:

注册一个标志,其值可以是 enum_values .

ARG: