python - 无法在 Keras 的 ImageDataGenerator 中将 preprocess_input 用作预处理函数

Question

当尝试提供从 CNN 包(例如 keras.applications.resnet)导入的 preprocess_input 时，模型会在尝试适应生成的数据时引发错误:

‘JpegImageFile’ object is not subscriptable

代码:

datagen = ImageDataGenerator(preprocessing_function=preprocess_input)

凯拉斯版本:2.1.5

Answer 1

通过将 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)))