Optimized

app.py

@@ -427,6 +427,16 @@ def generate_auto_masks(image, request_json):
         # Convert PIL to numpy
         image_array = np.array(image)
         H, W = image_array.shape[:2]
+
+        # Optional downscaling to keep masks smaller / faster
+        resize_longest = int(params.get("resize_longest", 0) or 0)
+        if resize_longest > 0 and max(H, W) > resize_longest:
+            scale = resize_longest / float(max(H, W))
+            new_w = max(1, int(W * scale))
+            new_h = max(1, int(H * scale))
+            print(f"Resizing image from {W}x{H} to {new_w}x{new_h} for auto masks...")
+            image_array = np.array(Image.fromarray(image_array).resize((new_w, new_h)))
+            H, W = image_array.shape[:2]
         
         print(f"Generating automatic masks for image of size {W}x{H}...")