python - 如何在 Tensorflow 中将灰度图像转换为 RGB？

Question

我有一个在这个输入形状上训练的 CNN 模型

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

在使用 skimage 处理我的图像后，我有以下形状

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

在这种情况下我该怎么办？

标签： pythontensorflowimage-processingkerastensor

Answer 1

我找到了这个解决方案：

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