TB_image_log.py

  # Create a TensorBoard writer for sample images
  image_writer=tf.summary.create_file_writer(logdir=image_log)
  # Generate an image grid in a format that can be used by TensorBoard
  rnd_imgs=image_funcs.create_TB_image_grid(dataset=ds,
                                            no_log_images=no_log_images, 
                                            log_images_split=log_images_split)

  # Add it to the TensorBoard logs
  with image_writer.as_default():
      tf.summary.image(name=rnd_imgs[0], data=rnd_imgs[2], 
                      max_outputs=no_log_images, step=0, 
                      description=rnd_imgs[1])