Как я узнаю, сколько фотографий создано с использованием ImageDataGeneration и fit_generator ()?

Question

Я использую ImageDataGenerator от Keras для реализации дополнения данных в Python. Я использую это следующим образом:

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

#dodanie augmentacji danych dla danych walidacyjnych
valid_datagen = ImageDataGenerator(rescale=1./255)

#pobranie danych treningowych
train_generator = train_datagen.flow_from_directory(directory=train_dir,
                                                   target_size=(224,224),
                                                   batch_size=32,
                                                   class_mode='binary')

#pobranie danych walidacyjnych
valid_generator = valid_datagen.flow_from_directory(directory=valid_dir,
                                                   target_size=(224,224),
                                                   batch_size=32,
                                                   class_mode='binary')

И затем я тренирую модель:

history = model.fit_generator(generator=train_generator,
                             steps_per_epoch=400,
                             epochs=5,    # 100
                             validation_data=valid_generator,
                             validation_steps=160, callbacks=[checkpt])

Как я узнаю, сколько изображений генерируется во время тренировки? Я имею в виду, есть ли способ узнать, сколько фотографий было обучено модели?

Answer 1

Если я не совсем не прав, ваши valid_datagen и train_datagen производят 32 изображения в одном go (определенном через batch_size), и вы активируете его 400 раз в каждую эпоху (steps_per_epoch), и это 5 раз (5 epochs). Это приводит к 32 * 400 * 5 = 64.000 обучающих изображений . Для проверки вы используете 32 * 160 = 5.120 Validation-Images . Но не исключено, что некоторые из сгенерированных изображений равны друг другу. Каждая партия имеет случайно сгенерированные изображения, поэтому существует вероятность репликации. Поправь меня, если я ошибаюсь, но так и должно быть