gpt4 book ai didi

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

转载 作者:太空宇宙 更新时间:2023-11-04 09:43:51 25 4
gpt4 key购买 nike

当尝试提供从 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/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com