阅读(6.5k) 书签 (0)

TensorFlow函数教程:tf.io.TFRecordOptions

2019-02-26 16:52 更新

tf.io.TFRecordOptions函数

类 TFRecordOptions

别名:

  • 类 tf.io.TFRecordOptions
  • 类 tf.python_io.TFRecordOptions

定义在:tensorflow/python/lib/io/tf_record.py。

用于操作TFRecord文件的选项。

__init__

__init__(
    compression_type=None,
    flush_mode=None,
    input_buffer_size=None,
    output_buffer_size=None,
    window_bits=None,
    compression_level=None,
    compression_method=None,
    mem_level=None,
    compression_strategy=None
)

创建一个TFRecordOptions实例。

当compression_type不是None时,选项仅影响TFRecordWriter。可以在zlib_compression_options.h和zlib手册中找到文档,详细信息和默认值。将选项保留为None,允许C ++设置合理的默认值。

参数:

  • compression_type:TFRecordCompressionType或None。
  • flush_mode:flush模式或None,默认值:Z_NO_FLUSH。
  • input_buffer_size:int或None。
  • output_buffer_size:int或None。
  • window_bits:int或None。
  • compression_level:0到9,或None。
  • compression_method:压缩方法或None。
  • mem_level:1到9,或None。
  • compression_strategy:策略或None。默认值:Z_DEFAULT_STRATEGY。

返回:

一个TFRecordOptions对象。

可能引发的异常:

  • ValueError:如果compression_type无效。

方法

get_compression_type_string

@classmethod
get_compression_type_string(
    cls,
    options
)

将各种选项类型转换为统一字符串。

参数:

  • options:TFRecordOption,TFRecordCompressionType或字符串。

返回:

压缩类型为字符串(例如'ZLIB','GZIP'或'')。

可能引发的异常:

  • ValueError:如果compression_type无效。

类成员

compression_type_map