Best Practice to log images in vision tasks
What is the best practice to log images?
Is there a standard procedure to log output images from the validation set to any kind of logger (e.g. Tensorboard)?
Usually, I like to log a number of outputs of say over the epochs to see how the prediction evolves.
Moreover, I pick a number of random samples and log them.
I am not quite sure how to do this with Pytorch Lightning and whether there is a common way to do it.
If not maybe I could help? My suggestion would be

Add a Callback for logging images
Get the indices of the samples one wants to log
Cache these samples in validation_step
Let the Callback log these images in on_epoch_end method