From 6c711e85bd8e39c1c15e8f26006b83bc40d04cf3 Mon Sep 17 00:00:00 2001 From: cmoyates Date: Sun, 1 Mar 2026 06:47:22 -0330 Subject: [PATCH] fix: remove double normalize_rgb in engine preprocessing preprocess() already calls normalize_rgb() internally; calling it before passing to preprocess() applied ImageNet normalization twice. Co-Authored-By: Claude Opus 4.6 --- src/corridorkey_mlx/engine.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/corridorkey_mlx/engine.py b/src/corridorkey_mlx/engine.py index 7e1cf25..209b805 100644 --- a/src/corridorkey_mlx/engine.py +++ b/src/corridorkey_mlx/engine.py @@ -17,7 +17,6 @@ from PIL import Image from corridorkey_mlx.inference.pipeline import load_model from corridorkey_mlx.io.image import ( - normalize_rgb, postprocess_alpha, postprocess_foreground, preprocess, @@ -141,7 +140,7 @@ class CorridorKeyMLXEngine: mask_f32 = np.asarray(mask_pil, dtype=np.float32)[:, :, np.newaxis] / 255.0 # -- preprocess (ImageNet norm + concat) -> (1, H, W, 4) NHWC -- - x = preprocess(normalize_rgb(rgb_f32), mask_f32) + x = preprocess(rgb_f32, mask_f32) # -- forward -- outputs = self._model(x)