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

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

  •  1
  • user288609  · 技术社区  · 6 年前

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

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

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

    1 回复  |  直到 6 年前
        1
  •  4
  •   wicke    6 年前

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

    从docstring:

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

    ARG:

    • 名称 : str ,标志名。
    • 违约 : str|None ,标志的默认值。
    • 枚举值 : [str] ,一个字符串的非空列表,其中包含标记的可能值。
    • 帮助 : STR ,帮助消息。