from keras_preprocessing.image import ImageDataGenerator
validation_generator = valid_datagen.flow_from_dataframe(val_data,
directory='./tiny-imagenet-200/val/images/',
x_col='Files',y_col='Class',target_size=(64,64),
color_mode='rgb',class_mode='categorical',batch_size=256,
shuffle=True,seed=42)
运行上述代码出现:AttributeError: 'ImageDataGenerator' object has no attribute 'flow_from_dataframe
我的第一感觉是keras版本问题,然后更新一下keras试一下!
方法一:更新keras版本
pip install --upgrade keras
pip install --upgrade keras=2.1.0(指定版本)
然而对于我的问题还是没解决,可能是其他原因!
方法二:重新安装keras-preprocessing(先卸载之前的,再重新安装)
(1)
1、pip uninstall keras-preprocessing
2、pip install keras-preprocessing
或者(这个没能解决我的问题!我的是用上面的方式解决的)
(2)
pip install git+https://github.com/keras-team/keras-preprocessing.git
然后再按照下面方式调用
from keras_preprocessing.image import ImageDataGenerator