Upload Colony_Analyzer_AI2_HF.py

Colony_Analyzer_AI2_HF.py

@@ -38,28 +38,36 @@ def pad(img_np, tw=2048, th=1536):
 
 
 #this is the huggingface version
+import numpy as np
+from PIL import Image
+
 def cut_img(img, patch_size=512):
-    img_map = {}
+    """
+    Cuts an image (PIL or numpy array) into non-overlapping patch_size x patch_size tiles.
+    Returns img_map, number of rows, number of cols.
+    """
+    # If it's a PIL image, convert to numpy array
     if isinstance(img, Image.Image):
-        width, height = img.size
         img_np = np.array(img)
+        height, width = img_np.shape[:2]
     else:
         img_np = img
         height, width = img_np.shape[:2]
+
     i_num = height // patch_size
     j_num = width // patch_size
+    img_map = {}
     count = 1
     for i in range(i_num):
         for j in range(j_num):
-            cropped_img = img.crop((
-                patch_size * j,
-                patch_size * i,
-                patch_size * (j + 1),
-                patch_size * (i + 1)
-            ))
+            cropped_img = img_np[
+                patch_size * i: patch_size * (i + 1),
+                patch_size * j: patch_size * (j + 1),
+                ...
+            ]
             img_map[count] = cropped_img
             count += 1
-    return img_map, i_num, j_num  # Return rows and cols for stitching
+    return img_map, i_num, j_num
     
 import numpy as np