decode_jpeg

paddle.vision.ops. decode_jpeg ( x, mode='unchanged', name=None ) [源代码]

将 JPEG 图像解码为三维 RGB Tensor 或者 一维灰度 Tensor，解码格式可选，输出 Tensor 的值为 uint8 类型，介于 0 到 255 之间。

参数

  • x (Tensor) - 包含 JPEG 图像原始字节的一维 uint8 Tensor。

  • mode (str，可选) - 转换图像模式选择，默认值为 'unchanged'。

  • name (str，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

具有形状（imge_channels、image_height、image_width）的解码图像 Tensor。

代码示例

>>> import cv2
>>> import numpy as np
>>> import paddle
>>> paddle.device.set_device('gpu')

>>> fake_img = (np.random.random(
...             (400, 300, 3)) * 255).astype('uint8')
>>> cv2.imwrite('fake.jpg', fake_img)
>>> img_bytes = paddle.vision.ops.read_file('fake.jpg')
>>> img = paddle.vision.ops.decode_jpeg(img_bytes)
>>> print(img.shape)
[3, 400, 300]