diff --git a/src/utils/image.py b/src/utils/image.py index 8e9e19c..e0d3991 100644 --- a/src/utils/image.py +++ b/src/utils/image.py @@ -46,7 +46,7 @@ def get_pseudo_rgb(arr: np.ndarray, gamma: float = 0.3) -> np.ndarray: a3[a3 > p9999] = p9999 a3 /= a3.max() - return np.stack([a1, a1, a1], axis=0) + return np.stack([a1, np.zeros(a1.shape), np.zeros(a1.shape)], axis=0) # return np.stack([a1, a2, a3], axis=0) @@ -141,11 +141,12 @@ class Image: self._size_bytes = self.path.stat().st_size self._dtype = self._data.dtype - logger.info( - f"Successfully loaded image: {self.path.name} " - f"({self._width}x{self._height}, {self._channels} channels, " - f"{self._format.upper()})" - ) + if 0: + logger.info( + f"Successfully loaded image: {self.path.name} " + f"({self._width}x{self._height}, {self._channels} channels, " + f"{self._format.upper()})" + ) except Exception as e: logger.error(f"Error loading image {self.path}: {e}")