有没有办法在“纯”Keras中将图像从灰度转换为RGB

时间:2018-05-17 13:43:25

标签: python keras deep-learning

我想知道是否有办法使用“纯”Keras将图像从灰度转换为RGB(即不导入Tensorflow)。

我现在所做的是:

x_rgb = tf.image.grayscale_to_rgb(x_grayscale)

1 个答案:

答案 0 :(得分:1)

也许你会考虑这个"作弊" (因为keras.backend最终可能会在幕后调用Tensorflow),但这是一个解决方案:

from keras import backend as K

def grayscale_to_rgb(images, channel_axis=-1):
    images= K.expand_dims(images, axis=channel_axis)
    tiling = [1] * 4    # 4 dimensions: B, H, W, C
    tiling[channel_axis] *= 3
    images= K.tile(images, tiling)
    return images

(假设您的灰度图像的形状为B x H x W而不是B x H x W x 1;否则只需删除该函数的第一行)