基于keras-ImageDataGenerator的图像预处理

2024-04-28 23:04:44 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图使用预处理函数在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

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


Tags: 数据函数模型图像cropimagereturndef