如何在 Tensorflow 中将灰度图像转换为 RGB?

How do I transform a grayscale image to RGB in Tensorflow?

我有一个基于此输入形状训练的 CNN 模型

TensorShape([1, 128, 256, 3])

在使用 skimage 处理我的图像后,我得到了以下形状

TensorShape([1, 128,256, 1])

遇到这种情况我该怎么办?

我找到了这个解决方案:

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