此页面由 Cloud Translation API 翻译。
Switch to English

tfds.features.FeaturesDict

GitHub上查看源代码

复合FeatureConnector ;在每个功能dict都有自己的连接器。

规范特征的编码/解码方法将递归编码/解码每个子连接器上的构造给出。其他功能可以从这个类和调用超()为了得到嵌套容器继承。

例:

对于DatasetInfo:

 features = tfds.features.FeaturesDict({
    'input': tfds.features.Image(),
    'output': tf.int32,
})
 

在生成时间:

 for image, label in generate_examples:
  yield {
      'input': image,
      'output': label
  }
 

在tf.data.Dataset()时间:

 for example in tfds.load(...):
  tf_input = example['input']
  tf_output = example['output']
 

对于嵌套特征,所述FeaturesDict将在内部扁平化的特征和转换到tf.train.Example密钥。事实上,tf.train.Example原不支持嵌套功能,而tf.data.Dataset一样。但是,内部改造应该是对用户不可见。

例:

 tfds.features.FeaturesDict({
    'input': tf.int32,
    'target': {
        'height': tf.int32,
        'width': tf.int32,
    },
})
 

将在内部存储数据为:

 {
    'input': tf.io.FixedLenFeature(shape=(), dtype=tf.int32),
    'target/height': tf.io.FixedLenFeature(shape=(), dtype=tf.int32),
    'target/width': tf.io.FixedLenFeature(shape=(), dtype=tf.int32),
}
 

feature_dict(字典):含有实施例的特征的连接器词典。通过tf.data.Dataset作为返回的键应该对应于数据字典()。类型(tf.int32,...)和类型的字典将自动转换成FeatureConnector。

ValueError 如果给定的特征之一是无法识别

dtype 返回此FeatureConnector的D型(或D型的字典)。
shape 返回此FeatureConnector的形状(或形状的字典)。

方法

decode_batch_example

查看源代码

解码多种功能分批在一个tf.Tensor。

此功能用于包裹在解码特征tfds.features.Sequence()默认情况下,该功能适用decode_example使用每个单独的元素tf.map_fn 。然而,对于优化功能,可以覆盖此方法以应用自定义批处理解码。

ARGS
tfexample_data 相同tf.Tensor输入作为decode_example ,但与和序列长度附加的第一尺寸。

返回
tensor_data 张量或字典张量,所述tf.data.Dataset对象的输出

decode_example

查看源代码

解码连载的例子。

ARGS
serialized_example 嵌套dicttf.Tensor
decoders 的嵌套字典Decoder对象,其允许自定义解码。结构应符合功能结构,但只有定制的功能键需要存在。见导向以获得更多信息。

返回
example 嵌套dict包含解码嵌套例子。

decode_ragged_example

查看源代码

解码从tf.RaggedTensor嵌套功能。

此功能用于包裹在嵌套解码特征tfds.features.Sequence()默认情况下,该功能适用decode_batch_example对衣衫褴褛的张量的平值。为了优化,功能可以覆盖此方法以应用自定义批处理解码。

ARGS
tfexample_data tf.RaggedTensor包含嵌套编码例子的输入。

返回
tensor_data 经解码的tf.RaggedTensor的tf.data.Dataset对象的或字典张量,输出

encode_example

查看源代码

详情请参阅基类。

get_serialized_info

查看源代码

详情请参阅基类。

get_tensor_info

查看源代码

详情请参阅基类。

items

查看源代码

keys

查看源代码

load_metadata

查看源代码

详情请参阅基类。

save_metadata

查看源代码

详情请参阅基类。

values

查看源代码

__contains__

查看源代码

__getitem__

查看源代码

返回与该键关联功能。

__iter__

查看源代码

__len__

查看源代码