如何使用train_datagen.flow_from_directory进行该文件夹的数据直接读取
时间: 2024-05-13 11:20:09 浏览: 105
Keras 数据增强ImageDataGenerator多输入多输出实例
train_datagen.flow_from_directory函数可以用于从一个文件夹读取数据并进行数据增强,详情可以参考Keras官网文档。调用方式如下:
```
from keras.preprocessing.image import ImageDataGenerator
train_datagen = ImageDataGenerator(rescale=1./255)
train_generator = train_datagen.flow_from_directory(
'path/to/directory',
target_size=(150, 150),
batch_size=20,
class_mode='binary')
```
其中,'path/to/directory' 是要读取的文件夹路径,target_size 是要将图片缩放到的大小,batch_size 是每个批次的大小,class_mode 是分类方式,可以是 'categorical','binary' 或 'sparse'。
阅读全文