From b80cc680f0d0ad695abd2595f49eaef4e0784fad Mon Sep 17 00:00:00 2001 From: renwerx <132638531+renwerx@users.noreply.github.com> Date: Sat, 25 Oct 2025 12:31:15 +1100 Subject: [PATCH] Add size parameter to image_to_tensor function Added input size parameter to image_to_tensor function for standardising image dimensions. Check if images match required input dimensions and resize them if not before conversion. Avoids tensor mismatch errors. --- Gen_3D_Modules/MV_Adapter/mvadapter/utils/saving.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/Gen_3D_Modules/MV_Adapter/mvadapter/utils/saving.py b/Gen_3D_Modules/MV_Adapter/mvadapter/utils/saving.py index 3282772e..bc67e2e7 100644 --- a/Gen_3D_Modules/MV_Adapter/mvadapter/utils/saving.py +++ b/Gen_3D_Modules/MV_Adapter/mvadapter/utils/saving.py @@ -18,12 +18,17 @@ from .typing import * +# standardise input dimensions +from torchvision import transforms def image_to_tensor( - images: List[Image.Image], device: str = "cuda" + images: List[Image.Image], device: str = "cuda", size=(512, 512) ) -> torch.FloatTensor: tensors = [] for image in images: + image = image.convert("RGB")# Prepare for sizing + if image.size != size:# If not required dimensions + image = image.resize(size, Image.LANCZOS) # Resize to required dimensions tensor = torch.from_numpy(np.array(image)).float() / 255.0 tensor = tensor.to(device) tensors.append(tensor)