捕获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

然后，我打开图像并查看它们，或者阅读它们并将其打印到笔记本上。但我不喜欢它-它笨重。有没有一种方法可以直接捕获生成器生成的更改图像？我想将它们添加到数组中。

谢谢！