python - flow_from_directory 能否从 Keras 的同一目录中获取训练数据和验证数据？

Question

我从 here 得到了下面的例子.

train_datagen = ImageDataGenerator(
        rescale=1./255,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True)

test_datagen = ImageDataGenerator(rescale=1./255)

train_generator = train_datagen.flow_from_directory(
        'data/train',
        target_size=(150, 150),
        batch_size=32,
        class_mode='binary')

validation_generator = test_datagen.flow_from_directory(
        'data/validation',
        target_size=(150, 150),
        batch_size=32,
        class_mode='binary')

训练和验证有两个单独的目录。只是好奇我是否可以从同一目录而不是两个单独的目录中获取训练和验证数据？有什么例子吗？

Answer 1

您可以将 validation_split 参数(0 到 1 之间的数字)传递给 ImageDataGenerator 类实例，以将数据拆分为训练集和验证集:

generator = ImagaDataGenerator(..., validation_split=0.3)

然后将 subset 参数传递给 flow_from_directory 以指定训练和验证生成器:

train_gen = generator.flow_from_directory(dir_path, ..., subset='training')
val_gen = generator.flow_from_directory(dir_path, ..., subset='validation')

注意:如果您为 ImageDataGenerator 设置了扩充参数，那么通过使用此解决方案，训练图像和验证图像都将被扩充。