keras test生成器flow_from_directory应输入哪些参数
时间: 2023-04-05 07:01:21 浏览: 73
Keras test生成器flow_from_directory应输入以下参数:目录路径、目标图像大小、颜色模式、类别模式、batch_size、shuffle、seed、save_to_dir、save_prefix、save_format。
相关问题
keras如何用创建test生成器以及具体参数, 请提供详细代码
您好,以下是使用 Keras 创建测试生成器的示例代码:
```python
from keras.preprocessing.image import ImageDataGenerator
# 创建测试数据生成器
test_datagen = ImageDataGenerator(rescale=1./255)
# 加载测试数据
test_generator = test_datagen.flow_from_directory(
'test_data_dir',
target_size=(150, 150),
batch_size=32,
class_mode='binary')
# 使用测试生成器进行预测
model.predict_generator(test_generator, steps=50)
```
其中,`test_data_dir` 是测试数据所在的目录,`target_size` 是将图像调整为的大小,`batch_size` 是每个批次的大小,`class_mode` 是分类模式,可以是二进制或分类。
希望这个代码能够帮到您。
test_generator = test_datagen.flow_from_directory 'test_dir', # 测试集目录 target_size=(224, 224), batch_size=32, class_mode='c
这是Keras的ImageDataGenerator类中的一个方法,用于从指定目录中生成测试数据。test_datagen是一个ImageDataGenerator对象,test_dir是要生成测试数据的目录。使用flow_from_directory()方法可以去读取该目录下的图像文件,并进行预处理,返回一个生成器对象,可以用于模型的测试。