阅读(7k) 书签 (0)

TensorFlow函数教程:tf.io.decode_image

2019-02-22 16:14 更新

tf.io.decode_image函数

别名:

  • tf.image.decode_image
  • tf.io.decode_image
tf.io.decode_image(
    contents,
    channels=None,
    dtype=tf.dtypes.uint8,
    name=None
)

定义在:tensorflow/python/ops/image_ops_impl.py。

decode_bmp,decode_gif,decode_jpeg和decode_png的便捷函数。

检测图像是BMP,GIF,JPEG还是PNG,并执行适当的操作将输入字节string转换为dtype类型的Tensor。

注意:decode_gif返回一个4-D数组[num_frames, height, width, 3],而不是decode_bmp,decode_jpeg和decode_png,它们返回3-D数组[height,width,num_channels]。如果要将GIF文件与BMP,JPEG或PNG文件混合,请确保在构建图形时考虑到这一点。

参数:

  • contents:0-D string。编码图像字节。
  • channels:可选的int。默认为0。解码图像的颜色通道数。
  • dtype:返回的Tensor所需的DType。
  • name:操作的名称(可选)

返回:

dtype类型的Tensor,对于BMP,JPEG和PNG图像其shape为[height, width, num_channels],对于GIF图像,其shape为[num_frames, height, width, 3]。

可能引发的异常:

  • ValueError:不正确的通道数。