martin/object-segmentation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Testing more grayscale

Browse Source
This commit is contained in:
Martin Laasmaa
2025-12-19 12:02:11 +02:00
parent b0c05f0225
commit 2b0b48921e
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
src/utils/image.py
View File

@@ -46,7 +46,7 @@ def get_pseudo_rgb(arr: np.ndarray, gamma: float = 0.3) -> np.ndarray:
a3[a3 > p9999] = p9999 a3[a3 > p9999] = p9999
a3 /= a3.max() 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) # return np.stack([a1, a2, a3], axis=0)
@@ -141,11 +141,12 @@ class Image:
self._size_bytes = self.path.stat().st_size self._size_bytes = self.path.stat().st_size
self._dtype = self._data.dtype self._dtype = self._data.dtype
logger.info( if 0:
f"Successfully loaded image: {self.path.name} " logger.info(
f"({self._width}x{self._height}, {self._channels} channels, " f"Successfully loaded image: {self.path.name} "
f"{self._format.upper()})" f"({self._width}x{self._height}, {self._channels} channels, "
) f"{self._format.upper()})"
)
except Exception as e: except Exception as e:
logger.error(f"Error loading image {self.path}: {e}") logger.error(f"Error loading image {self.path}: {e}")