python - 使用 keras ImageDataGenerator 进行图像预处理

Question

我试图使用 preprocessing_function 在 ImageDataGenerator 中放置一个预处理函数。数据由图像组成。我制作了如下预处理图像。

def preprocessing(images):
    images = alignment(images)
    images = crop(images)
    return images

我已经用另一个图像检查了这个功能，它工作正常，但它无法预处理和训练模型。错误是 - “不支持的图像类型，必须是 8 位灰度或 RGB 图像。” . 错误指向对齐功能。

def alignment(image):
    gray_image = cv2.cvtColor(image ,cv2.COLOR_BGR2GRAY)
    # other stuffs

错误出现在对齐函数的第一行。请帮我 。

标签： pythontensorflowimage-processingkerasdeep-learning