Upload Colony_Analyzer_AI2_HF.py

Colony_Analyzer_AI2_HF.py

@@ -11,21 +11,14 @@ import cv2
 from PIL import Image
 
 
-def ensure_minimum_size(img, min_width=512, min_height=512):
-    width, height = img.size
-    if width < min_width or height < min_height:
-        return img.resize((max(width, min_width), max(height, min_height)), Image.BICUBIC)
-    return img
-
-def pad(img_np):
+def pad(img_np, tw=2048, th=1536):
     """
     Pads a numpy image (grayscale or RGB) to 2048x1536 (width x height) with white pixels.
     Pads at the bottom and right as needed.
     """
-    target_width, target_height = 2048, 1536
     height, width = img_np.shape[:2]
-    pad_bottom = max(0, target_height - height)
-    pad_right = max(0, target_width - width)
+    pad_bottom = max(0, th - height)
+    pad_right = max(0, tw - width)
     # Padding: (top, bottom, left, right)
     if img_np.ndim == 3:
         # Color image (H, W, 3)
@@ -267,7 +260,7 @@ def main(args):
     min_circ = args[2]
     do_necrosis = args[3]
     colonies = {}
-    img_map, i_num, j_num = cut_img(ensure_minimum_size(args[0], min_width=512, min_height=512))
+    img_map, i_num, j_num = cut_img(pad(args[0],512,512))
     for z in img_map:
         img_map[z] = eval_img(img_map[z])
     del z