如何在 keras 中进行自定义回调以在 VAE 训练中生成示例图像？

数据挖掘 Python 喀拉斯 自动编码器 生成模型 vae
2022-02-16 05:32:59

我正在 64*64 图像上训练一个简单的 VAE 模型，我想查看每个 epoch 或每几批生成的图像以查看进度。

当我训练模型时，我会等到训练完成后再查看结果。

我试图在 Keras 中创建一个自定义回调函数来生成图像并保存它但无法做到。有可能吗？我找不到类似的东西。

如果您将我推荐给解释如何这样做或向我展示示例的来源，那就太棒了

1个回答

我认为以下两个链接可以帮助您1、2。

第一个是教程，向您介绍如何在 TensorBoard 中显示图像。
如果您查看混淆矩阵上的部分，您应该找到一种方法来进行所需的回调，即涉及显示图像的回调。

第二个链接也是一个教程，向您展示如何使用 Keras 在 TensorFlow 中编写您自己的自定义回调。

我认为通过结合这两个教程，您应该对如何做您想做的事情有一个很好的了解。

其它你可能感兴趣的问题
上一篇在机器学习的背景下，退火意味着什么？ 下一篇推荐系统学习资源