def METHOD_NAME(self, epoch, logs=None):
    if logs is None:
        logs = {}
    super(TensorBoardImage, self).METHOD_NAME(epoch, logs)
    images = self.validation_data[0]  # 0 - data; 1 - labels
    img = (255 * images[0].reshape(28, 28)).astype('uint8')
    image = self.make_image(img)
    summary = tf.Summary(value=[tf.Summary.Value(tag='image', image=image)])
    self.writer.add_summary(summary, epoch)