我正在使用keras和fit_generator
来训练我的模型。我正在使用的当前模型是自动编码器-无法呈现所需的结果。因此,我想创建一个回调,说明每500个批次中的训练图像和groundtruth图像。因此,我想使用on_batch_begin
,但是不确定如何访问当前批次以创建tf.Summary.Image
。
任何人都可以引导我了解有关此的一些信息,或者知道如何获取当前批次。还是会在生成器中完成?我只是看不到如何附加一个回调。
答案 0 :(得分:0)
我一直找不到合适的解决方案。我已经向要添加文件的数组添加了要分析的文件。然后,我随机选择一张图像进行说明,并使用当前模型并在张量板上进行说明。它可以工作,但我希望它会更加优雅:)