keras - 如何在 Keras 中转换图像颜色空间？

Question

我正在将 RGB 彩色图像输入到使用 Keras 实现的神经网络。如何让 Keras 将图像转换为不同的颜色空间(例如 YUV、Lab 或某些灰度)？

我尝试使用 Lambda() 层，但收到错误:

model.add(Lambda(lambda x: cv2.cvtColor(x, cv2.COLOR_RGB2LAB), input_shape=(160, 320, 3)))

给了我

类型错误:src 不是 numpy 数组，也不是标量

我认为问题是 x 是一个张量，我不知道如何将其转换为 OpenCV 接受的东西。

如果我可以在 GPU 中完成它就更好了。例如。对于 Tensorflow，我将使用诸如 tf.image.rgb_to_hsv() 和 tf.image.rgb_to_grayscale() 等函数。

谢谢!

Answer 1

如果导入tensorflow，则可以在lambda中使用tf.image.rgb_to_hsv()函数:

def hsv_conversion(x):
    import tensorflow as tf    
    return tf.image.rgb_to_hsv(x)

model.add(Lambda(hsv_conversion, input_shape=(160, 320, 3)))