diff --git a/utils/loggers/__init__.py b/utils/loggers/__init__.py index 3d67e93..16d0348 100644 --- a/utils/loggers/__init__.py +++ b/utils/loggers/__init__.py @@ -131,10 +131,9 @@ class Loggers(): files = [(self.save_dir / f) for f in files if (self.save_dir / f).exists()] # filter if self.tb: - from PIL import Image - import numpy as np + import cv2 for f in files: - self.tb.add_image(f.stem, np.asarray(Image.open(f)), epoch, dataformats='HWC') + self.tb.add_image(f.stem, cv2.imread(str(f))[..., ::-1], epoch, dataformats='HWC') if self.wandb: self.wandb.log({"Results": [wandb.Image(str(f), caption=f.name) for f in files]})