当尝试提供从 CNN 包(例如 keras.applications.resnet)导入的 preprocess_input 时,模型会在尝试适应生成的数据时引发错误:
‘JpegImageFile’ object is not subscriptable
代码:
datagen = ImageDataGenerator(preprocessing_function=preprocess_input)
凯拉斯版本:2.1.5
最佳答案
通过将 Keras 降级到 2.1.4 来解决问题:
pip uninstall keras
pip install keras==2.1.4
或者通过使用 Lambda 层将预处理添加为模型的第一步:
model.add(keras.layers.Lambda(preprocess_input, name='preprocessing', input_shape=(224, 224, 3)))
关于python - 无法在 Keras 的 ImageDataGenerator 中将 preprocess_input 用作预处理函数,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50547382/