比较与tf.image.grayscale_to_rgb的功能差异

tf.image.grayscale_to_rgb

tf.image.grayscale_to_rgb(
    images,
    name=None
)

更多内容详见tf.image.grayscale_to_rgb。

mindspore.dataset.vision.ConvertColor

class mindspore.dataset.vision.ConvertColor(
    convert_mode
)

更多内容详见mindspore.dataset.vision.ConvertColor。

使用方式

TensorFlow：将图像从灰度图转换为RGB图。

MindSpore：转换图像的色彩空间，其中包括将灰度图转换为RGB图。

代码示例

# The following implements ConvertColor with MindSpore.
import numpy as np
import mindspore.dataset as ds
from mindspore.dataset.vision import ConvertMode

image = np.random.random((28, 28, 1)).astype(np.float32)
result = ds.vision.ConvertColor(ConvertMode.COLOR_GRAY2RGB)(image)
print(result.shape)
# (28, 28, 3)

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

image = tf.random.normal((28, 28, 1))
result = tf.image.grayscale_to_rgb(image)
print(result.shape)
# (28, 28, 3)