python - ImageData 生成器中的 validation_split 返回 0 个图像

Question

我试图通过将 validation_split 设置为某个分数来使用 Keras 中的 ImageDataGenerator 拆分我的数据(图像)。

这是我的代码:

# Generate batches of tensor image data with real-time data augmentation,  looped over in batches 
train_DataGen_Augmnt =ImageDataGenerator(
    rescale=1./255, 
    featurewise_center=True,
    validation_split=0.2,
    rotation_range=30,
    horizontal_flip=True,
)

# validation data not augmented!
Validation_DataGen = ImageDataGenerator(rescale=1./255)

# Flow training images in batches of 32 using train_datagen generator
train_generator = train_DataGen_Augmnt.flow_from_directory(
    base_dir,
    subset='training',   
    target_size=(150, 150),  
    batch_size=32,
    #save_to_dir='images/Agumented'
)

# Flow validation images in batches of 32 using test_datagen generator
validation_generator = Validation_DataGen.flow_from_directory(
    base_dir,
    subset='validation',
    target_size=(150, 150),
    batch_size=32,
)

拆分似乎在拆分数据时起作用，但仅显示了仅用于训练的拆分。

这是运行时的输出:

Found 89 images belonging to 3 classes.
Found 0 images belonging to 3 classes. 

Answer 1

使用与生成训练数据相同的生成器 train_DataGen_Augmnt 生成验证数据