我正在尝试使用 来了解数据增强ImageDataGenerator
,为了深入了解它,我创建了一个小型数据集。我save_to_dir
用来查看图像是如何增强的
主目录:/图像
火车目录:/图像/火车
dog_directory : /Images/train/dogs #10 狗的图像
cat_directory : /Images/train/cats #10 张猫的图片
save_to_dir : /Augmented_images
=> 这是代码
train_datagen = ImageDataGenerator(
rescale=1./255,
rotation_range=40,
width_shift_range=0.2,
height_shift_range=0.2,
shear_range=0.2,
zoom_range=0.2,
horizontal_flip=True,
fill_mode='nearest')
train_generator = train_datagen.flow_from_directory(
train_directory,
target_size=(150, 150),
batch_size=5,
class_mode='binary')
history = model.fit_generator(
train_generator,
steps_per_epoch=4, # 20(number of total images)/5 (batch_size)
epochs=1,)
所以,我的问题是,如果有 20 张图片,batch_size=5, 1 epoch,steps_per_epoch=4 因此,“Augmented_images”目录中的图片数量应该是 20。但是,它显示了 45 张图片。如何?这是否意味着即使steps_per_epoch = 4,也训练了9次迭代?