I'm training a simple VAE model on 64*64 images and I would like to see the images generated after every epoch or every couple batches to see the progress.

when I train the model I wait until the training is done and then I look at the results.

I tried to make a custom callback function in Keras that generates an image and saves it but couldn't do it. is it even possible? I couldn't find anything like it.

it would be awesome if you refer me to a source that explains how to do so or show me an example


1 Answer 1


I think the two following links could help you 1, 2.

The first one is a tutorial, which introduces you how to display images in TensorBoard.
If you look at the part on the confusion matrix, you should find a way to make your desired callback, i.e. a callback which involves displaying an image.

The second link is also a tutorial that shows you how to write your very own custom callback in TensorFlow with Keras.

I think by combining both tutorial you should have a pretty good idea how to do what you want.


Your Answer

By clicking “Post Your Answer”, you agree to our terms of service and acknowledge you have read our privacy policy.

Not the answer you're looking for? Browse other questions tagged or ask your own question.