Update controlnet_module.py

controlnet_module.py

@@ -151,6 +151,15 @@ class ControlNetProcessor:
             with torch.no_grad():
                 outputs = self.sam_model(**inputs)
 
+
+            # DEBUG: Dimensionen prüfen
+            print(f"🔍 Original image size: {image.size}")  # z.B. (1024, 768)
+            print(f"🔍 Processed image size: {inputs['pixel_values'].shape}")  # z.B. torch.Size([1, 3, 1024, 1024])
+            print(f"🔍 Output masks shape: {outputs.pred_masks.shape}")  # KRITISCH: z.B. torch.Size([1, 2, 256, 256])
+            print(f"🔍 Original_sizes parameter (in inputs): {inputs.get('original_sizes', 'NOT FOUND')}")
+            print(f"🔍 Reshaped_input_sizes parameter (in inputs): {inputs.get('reshaped_input_sizes', 'NOT FOUND')}")
+
+
             # 5. Maske extrahieren und verarbeiten
             # outputs.pred_masks enthält die Masken-Logits
             # post_process_masks stellt die Originalgröße wieder her