Keras flow_from_directory限制示例数量

from keras.preprocessing.image import ImageDataGenerator

# 创建ImageDataGenerator对象
datagen = ImageDataGenerator()

# 设置数据加载和预处理的参数
batch_size = 32
target_size = (224, 224)

# 从文件夹中加载训练集数据
train_generator = datagen.flow_from_directory(
    directory='path/to/train_folder',
    target_size=target_size,
    batch_size=batch_size,
    subset='training'
)

# 从文件夹中加载验证集数据
validation_generator = datagen.flow_from_directory(
    directory='path/to/train_folder',
    target_size=target_size,
    batch_size=batch_size,
    subset='validation'
)

# 从文件夹中加载测试集数据
test_generator = datagen.flow_from_directory(
    directory='path/to/test_folder',
    target_size=target_size,
    batch_size=batch_size
)