tf.image.grayscale_to_rgb

Преобразует один или несколько изображений из оттенков серого в RGB.

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

Выводит тензор с таким же DType и рангом, как и images. Размер последнего измерения выходного значения равен 3, содержащему значение RGB пикселей. Последнее измерение входных изображений должно быть размером 1.

original = tf.constant([[[1.0], [2.0], [3.0]]])
converted = tf.image.grayscale_to_rgb(original)
print(converted.numpy())
[[[1. 1. 1.]
  [2. 2. 2.]
  [3. 3. 3.]]]