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

RGB

Browse Source
This commit is contained in:
Martin Laasmaa
2025-12-19 13:31:24 +02:00
parent 2bcd18cc75
commit 2fd9a2acf4
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/utils/image.py
View File

@@ -46,8 +46,8 @@ 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, np.zeros(a1.shape), np.zeros(a1.shape)], 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)
class ImageLoadError(Exception): class ImageLoadError(Exception):