0

我正在尝试使用 来了解数据增强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次迭代？

4

0 回答 0