Как я узнаю, что мой ImageDataGenerator работает?

Question

Я пытаюсь использовать ImageDataGenerator для искусственного увеличения общего количества тренировочных образов. Мои изображения составляют около 200 000, а общее количество изображений, использованных за эпоху при использовании model.fit_generator , составляет около 5000+, что является тем же числом, когда я использую model.fit . Могу ли я увидеть общее количество "сгенерированных" изображений? Как узнать, работает ли ImageDataGenerator?

Answer 1

Вы можете построить какое-нибудь изображение генератора данных:

#catch a few image of train_generator.
x_batch, y_batch = next(train_generator)

plt.figure(figsize=(12, 9))
for k, (img, lbl) in enumerate(zip(x_batch, y_batch)):
    plt.subplot(4, 8, k+1)#4 rows with 8 images.
    plt.imshow((img + 1) / 2)
    plt.axis('off')

Количество образцов - следующий код:

print (train_generator.samples)