Keras：无法访问 on_batch_end 回调中的训练图像

Question

我正在使用带有 TensorFlow 后端的 Keras 训练 CNN，使用 imgaug 进行图像增强。

我也在使用 Tensorboard 来可视化训练进度和结果。

由于 imgaug 正在对输入图像应用（随机）转换，我想将（一些）增强图像发送到 Tensorboard，以便我可以可视化它们并验证一切是否正确（例如：检查是否我应用了太大的翻译，或者图像过于模糊）。

为此，我创建了一个自定义 Keras 回调，并尝试在 on_batch_end 方法中输入我的逻辑。我可以将图像发送到张量板，但找不到可以访问增强输入图像的位置。关于如何实现这一点的任何提示？

提前致谢

Answer 1

最好通过简单地从生成器中获取图像来进行外部训练。

如果是普通的generator:

for i in range(numberOfBatches):
    x,y = next(generator)
    #plot, print, etc. with the batches    

如果是keras.utils.Sequence:

for i in range(len(generator)):
    x,y = generator[i]
    #plot, print, etc. with the batches