关于 tf.app.flags.DEFINE_enum 的使用

use*_*609 1 tensorflow

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

tf.app.flags.DEFINE_enum('image_format', 'png', ['jpg', 'jpeg', 'png'],
                         'Image format.')
Run Code Online (Sandbox Code Playgroud)

但我找不到 的用法定义tf.app.flags.DEFINE_enum,它有什么用?

wic*_*cke 5

这将创建一个名为 的命令行参数image_format,默认值为png,并且可以采用值pngjpegjpg。在程序帮助字符串中,该标志将被简单地解释为“图像格式”。

从文档字符串:

注册一个标志,其值可以是来自 的任何字符串enum_values

参数:

  • name : str,标志名称。
  • default : str|None,标志的默认值。
  • enum_values[str]一个非空的字符串列表,其中包含标志的可能值。
  • help : str,帮助消息。
  • ...