Keras:无法访问“批处理结束呼叫”中的训练图像

Question

我正在训练一个CNN，使用Keras和TensorFlow后端，使用imgauge进行图像增强。在

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

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

为此，我创建了一个自定义的Keras回调，并尝试在on_batch_end方法中输入我的逻辑。我可以把图像发送到tensorboard，但是找不到可以访问增强输入图像的位置。有什么秘诀可以帮助你实现这个目标吗？在

提前谢谢

Answer 1

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

如果是常规的generator：

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

如果是keras.utils.Sequence：

^{pr2}$