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

时间:2018-06-03 21:15:51

标签: tensorflow

在程序中,我看到了以下代码段

models

但我无法找到tf.app.flags.DEFINE_enum('image_format', 'png', ['jpg', 'jpeg', 'png'], 'Image format.') 的使用定义,它用于什么?

1 个答案:

答案 0 :(得分:3)

这会生成一个名为image_format的命令行参数,默认值为png,并且可以采用值pngjpeg或{{1} }。在程序帮助字符串中,标志将简单地解释为"图像格式。"。

来自docstring:

注册一个标志,其值可以是jpg中的任何字符串。

参数数量:

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