diff --git a/src/utils/image.py b/src/utils/image.py index 0f93969..217b0de 100644 --- a/src/utils/image.py +++ b/src/utils/image.py @@ -47,6 +47,7 @@ def get_pseudo_rgb(arr: np.ndarray, gamma: float = 0.3) -> np.ndarray: a3 /= a3.max() # 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) diff --git a/src/utils/ultralytics_16bit_patch.py b/src/utils/ultralytics_16bit_patch.py index 29f6cc5..0ea98e2 100644 --- a/src/utils/ultralytics_16bit_patch.py +++ b/src/utils/ultralytics_16bit_patch.py @@ -76,8 +76,8 @@ def apply_ultralytics_16bit_tiff_patches(*, force: bool = False) -> None: # Ensure contiguous array for downstream OpenCV ops. # logger.info(f"Loading with monkey-patched imread: {filename}") - arr *= 2**16 - 1 - arr = arr.astype(np.uint16) + arr *= 2**8 - 1 + arr = arr.astype(np.uint8) return np.ascontiguousarray(arr) # logger.info(f"Loading with original imread: {filename}")