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

tfds.features.Image

GitHub上查看源代码

FeatureConnector的图像。

继承: FeatureConnector

期间_generate_examples ,特征连接器接受作为输入的任何一个:

  • str :路径到{BMP,GIF,JPEG,PNG}图片(例如: /path/to/img.png )。
  • np.array :三维np.uint8表示图像的阵列。
  • 含有PNG或JPEG编码的图像的字符串的文件对象(例如: io.BytesIO(encoded_img_bytes)

输出:

tf.Tensor类型的tf.uint8和形状[height, width, num_channels]为BMP,JPEG和PNG图像和形状[num_frames, height, width, 3]为GIF图像。

例:

 features=features.FeaturesDict({
    'input': features.Image(),
    'target': features.Image(shape=(None, None, 1),
                               encoding_format='png'),
})
 
  • 生成过程中:
 yield {
    'input': 'path/to/img.jpg',
    'target': np.ones(shape=(64, 64, 1), dtype=np.uint8),
}
 

shape 整数或无的元组,解码图像的形状。对于GIF图像:(num_frames,高度,宽度,通道= 3)。 num_frames,高度和宽度可以是无。对于其他的图像:(高度,宽度,信道)。高度和宽度可以是无。见tf.image.encode_*用于在通道参数文档。默认为(无,无,3)。
dtype tf.uint16或tf.uint8(默认)。 tf.uint16只能与PNG encoding_format使用
encoding_format 'JPEG' 或 'PNG'(缺省)。格式到磁盘上的序列化np.ndarray图像。如果图像是从{BMG,GIF,JPEG,PNG}文件加载,这个参数将被忽略,并且被用于文件原始编码。

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

查看源代码

重建从TF示例的图像。

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

查看源代码

给定的图像转换为可转换为TF例如一个字典。

get_serialized_info

查看源代码

返回的特征的形状/ D型编码(对于适配器)之后。

FileAdapter然后使用这些信息来写入数据在磁盘上。

该函数表示该功能是如何对文件编码的内部。该DatasetBuilder都写在磁盘上,tf.train.Example原。

例如:

 return {
    'image': tfds.features.TensorInfo(shape=(None,), dtype=tf.uint8),
    'height': tfds.features.TensorInfo(shape=(), dtype=tf.int32),
    'width': tfds.features.TensorInfo(shape=(), dtype=tf.int32),
}
 

FeatureConnector这不是容器应直接返回功能原型:

 return tfds.features.TensorInfo(shape=(64, 64), tf.uint8)
 

如果没有定义,则重新调谐值被自动从推导get_tensor_info功能。

返回
features 任一特征原对象的字典,或特征原始对象

get_tensor_info

查看源代码

返回功能的tf.Tensor D型/形状。

这将返回张量D型/形状,如通过返回通过.as_dataset tf.data.Dataset对象。

例如:

 return {
    'image': tfds.features.TensorInfo(shape=(None,), dtype=tf.uint8),
    'height': tfds.features.TensorInfo(shape=(), dtype=tf.int32),
    'width': tfds.features.TensorInfo(shape=(), dtype=tf.int32),
}
 

FeatureConnector这不是容器应直接返回功能原型:

 return tfds.features.TensorInfo(shape=(256, 256), dtype=tf.uint8)
 

返回
tensor_info 任一个的字典tfds.features.TensorInfo对象或tfds.features.TensorInfo

load_metadata

查看源代码

详情请参阅基类。

save_metadata

查看源代码

详情请参阅基类。

set_dtype

查看源代码

更新D型。

set_encoding_format

查看源代码

更新编码格式。

set_shape

查看源代码

更新的形状。