Захватить вывод ImageDataGenerator без сохранения изображений на диск?

Question

Я использую увеличение изображения для обучения модели, применяя преобразования, такие как изменение яркости и цвета.Мне нравится предварительно просматривать эффекты дополнений, прежде чем вводить их в действие.Обычно я делаю это так:

from keras_preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator(horizontal_flip = True,
                             fill_mode = "nearest",
                             zoom_range = 0.3,
                             rotation_range=360)

i = 0

for batch in datagen.flow_from_directory(directory='./my_images/,
                                         batch_size = 1,
                                         save_to_dir='.',
                                         save_prefix='aug',
                                         save_format='jpeg'):
    i += 1
    if i > 2:
        break  # Yields two images

Затем я открываю изображения и смотрю на них, или читаю их и распечатываю в своей записной книжке.Но мне это не нравится - это неуклюже.Есть ли способ напрямую захватить измененные изображения, полученные моим генератором?Я хотел бы добавить их в массив.

Спасибо!