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

Another test

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

1
src/utils/image.py
View File

@@ -47,6 +47,7 @@ def get_pseudo_rgb(arr: np.ndarray, gamma: float = 0.3) -> np.ndarray:
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([a2, 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)

4
src/utils/ultralytics_16bit_patch.py
View File

@@ -76,8 +76,8 @@ def apply_ultralytics_16bit_tiff_patches(*, force: bool = False) -> None:
# Ensure contiguous array for downstream OpenCV ops. # Ensure contiguous array for downstream OpenCV ops.
# logger.info(f"Loading with monkey-patched imread: {filename}") # logger.info(f"Loading with monkey-patched imread: {filename}")
arr *= 2**16 - 1 arr *= 2**8 - 1
arr = arr.astype(np.uint16) arr = arr.astype(np.uint8)
return np.ascontiguousarray(arr) return np.ascontiguousarray(arr)
# logger.info(f"Loading with original imread: {filename}") # logger.info(f"Loading with original imread: {filename}")