From 483dad82879b3fe21d591dfacecfece5f3e41ab5 Mon Sep 17 00:00:00 2001 From: Andrew White Date: Mon, 27 Jul 2026 20:48:17 -0500 Subject: [PATCH] fix: rgb crops batch dimension instead of H/W --- dust3r/utils/image.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/dust3r/utils/image.py b/dust3r/utils/image.py index 6312a34..a30a62d 100644 --- a/dust3r/utils/image.py +++ b/dust3r/utils/image.py @@ -51,8 +51,11 @@ def rgb(ftensor, true_shape=None): elif ftensor.ndim == 4 and ftensor.shape[1] == 3: ftensor = ftensor.transpose(0, 2, 3, 1) if true_shape is not None: - H, W = true_shape - ftensor = ftensor[:H, :W] + H, W = true_shape[-2:] + if ftensor.ndim == 4: + ftensor = ftensor[:, :H, :W] + else: + ftensor = ftensor[:H, :W] if ftensor.dtype == np.uint8: img = np.float32(ftensor) / 255 else: