Keras ImageDataGenerator 数据增强与生成器使用示例

test_datagen = ImageDataGenerator(rescale=1.0/255)  # 创建一个ImageDataGenerator对象，用于对图像进行预处理，将像素值缩放到0-1之间
test_generator = test_datagen.flow_from_directory(
                                                    'test_dataset_path',            # 测试集目录的路径
                                                    target_size=(150, 150),       # 将输入的图像大小调整为 (150, 150)
                                                    batch_size=32,                # 单个批次包含的样本数
                                                    class_mode='categorical',     # 测试集的标签为分类
                                                    shuffle=True)                 # 打乱样本的顺序，以增加模型的泛化能力