 GitHub上查看源代码 |
复合FeatureConnector ;在每个功能dict都有自己的连接器。
tfds.features.FeaturesDict(
feature_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),
}
ARGS | |
|---|---|
| feature_dict（字典）：含有实施例的特征的连接器词典。通过tf.data.Dataset作为返回的键应该对应于数据字典（）。类型（tf.int32，...）和类型的字典将自动转换成FeatureConnector。 |
加薪 | |
|---|---|
ValueError | 如果给定的特征之一是无法识别 |
属性 | |
|---|---|
dtype | 返回此FeatureConnector的D型（或D型的字典）。 |
shape | 返回此FeatureConnector的形状（或形状的字典）。 |
方法
decode_batch_example
decode_batch_example(
tfexample_data
)
解码多种功能分批在一个tf.Tensor。
此功能用于包裹在解码特征tfds.features.Sequence()默认情况下，该功能适用decode_example使用每个单独的元素tf.map_fn 。然而，对于优化功能，可以覆盖此方法以应用自定义批处理解码。
| ARGS | |
|---|---|
tfexample_data | 相同tf.Tensor输入作为decode_example ，但与和序列长度附加的第一尺寸。 |
| 返回 | |
|---|---|
tensor_data | 张量或字典张量，所述tf.data.Dataset对象的输出 |
decode_example
decode_example(
serialized_example, decoders=None
)
解码连载的例子。
| ARGS | |
|---|---|
serialized_example | 嵌套dict的tf.Tensor |
decoders | 的嵌套字典Decoder对象，其允许自定义解码。结构应符合功能结构，但只有定制的功能键需要存在。见导向以获得更多信息。 |
| 返回 | |
|---|---|
example | 嵌套dict包含解码嵌套例子。 |
decode_ragged_example
decode_ragged_example(
tfexample_data
)
解码从tf.RaggedTensor嵌套功能。
此功能用于包裹在嵌套解码特征tfds.features.Sequence()默认情况下，该功能适用decode_batch_example对衣衫褴褛的张量的平值。为了优化，功能可以覆盖此方法以应用自定义批处理解码。
| ARGS | |
|---|---|
tfexample_data | tf.RaggedTensor包含嵌套编码例子的输入。 |
| 返回 | |
|---|---|
tensor_data | 经解码的tf.RaggedTensor的tf.data.Dataset对象的或字典张量，输出 |
encode_example
encode_example(
example_dict
)
详情请参阅基类。
get_serialized_info
get_serialized_info()
详情请参阅基类。
get_tensor_info
get_tensor_info()
详情请参阅基类。
items
items()
keys
keys()
load_metadata
load_metadata(
data_dir, feature_name=None
)
详情请参阅基类。
save_metadata
save_metadata(
data_dir, feature_name=None
)
详情请参阅基类。
values
values()
__contains__
__contains__(
k
)
__getitem__
__getitem__(
key
)
返回与该键关联功能。
__iter__
__iter__()
__len__
__len__()