比较与tf.io.decode_image的功能差异

查看源文件

tf.io.decode_image

tf.io.decode_image(
    contents,
    channels=None,
    dtype=tf.dtypes.uint8,
    name=None,
    expand_animations=True
)

更多内容详见tf.io.decode_image

mindspore.dataset.vision.Decode

class mindspore.dataset.vision.Decode(
    rgb=True
)

更多内容详见mindspore.dataset.vision.Decode

使用方式

TensorFlow:将图像字节码解码为指定通道数和数据类型的图像,支持解码动态图。

MindSpore:将图像字节码解码为RGB图像。

代码示例

# The following implements Decode with MindSpore.
import numpy as np
import mindspore.dataset as ds

image = np.fromfile("/tmp/file.jpg", dtype=np.uint8)
result = ds.vision.Decode()(image)

# The following implements decode_image with TensorFlow.
import tensorflow as tf

raw = tf.io.read_file("/tmp/file.jpg")
result = tf.io.decode_image(raw, channels=3)