使用生成器在Tensorboard中说明批处理

时间:2019-05-09 12:29:10

标签: keras tensorboard

我正在使用keras和fit_generator来训练我的模型。我正在使用的当前模型是自动编码器-无法呈现所需的结果。因此,我想创建一个回调,说明每500个批次中的训练图像和groundtruth图像。因此,我想使用on_batch_begin,但是不确定如何访问当前批次以创建tf.Summary.Image

任何人都可以引导我了解有关此的一些信息,或者知道如何获取当前批次。还是会在生成器中完成?我只是看不到如何附加一个回调。

1 个答案:

答案 0 :(得分:0)

我一直找不到合适的解决方案。我已经向要添加文件的数组添加了要分析的文件。然后,我随机选择一张图像进行说明,并使用当前模型并在张量板上进行说明。它可以工作,但我希望它会更加优雅:)