TensorFlow 2.0 Keras: How to write image summaries for TensorBoard
Except providing an answer to your question I will make the code more TF2.0-like. If you have any questions/need clarification, please post a comment down below. 1. Loading data I would advise to use Tensorflow Datasets library. There is absolutely no need to load data in numpy and transform it to tf.data.Dataset if one can … Read more