- fixed image normalization in tensorboard

inferno/trainers/callbacks/logging/tensorboard.py

@@ -341,8 +341,15 @@ def log_images(self, tag, images, step, image_format='CHW'):
             # tensorboardX borks if the number of image channels is not 3
             if image.shape[-1] == 1:
                 image = image[..., [0, 0, 0]]
+            image = self._normalize_image(image)
             self.writer.add_image(tag, img_tensor=image, global_step=step)
 
+    @staticmethod
+    def _normalize_image(image):
+        normalized_image = image - image.min()
+        normalized_image = normalized_image / normalized_image.max()
+        return normalized_image
+
     def log_histogram(self, tag, values, step, bins=1000):
         """Logs the histogram of a list/vector of values."""
         # TODO