decrease number of crops

app.py

@@ -71,8 +71,6 @@ from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamP
 model = sam_model_registry["default"](checkpoint="./sam_vit_h_4b8939.pth")
 model.to(device)
 
-predictor = SamPredictor(model)
-
 mask_generator = SamAutomaticMaskGenerator(model)
 
 import gradio as gr
@@ -106,7 +104,7 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
 
     print(cropped_image.shape)
 
-    split_num = 3
+    split_num = 2
 
     x_inc = int(cropped_image.shape[0]/split_num)
     y_inc = int(cropped_image.shape[1]/split_num)
@@ -127,7 +125,6 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
             # plt.figure()
             # plt.imshow(small_image)
             # plt.axis('on')
-            mask_generator.predictor.set_image(small_image)
             progress(0, desc=f"Generating masks for crop {r*split_num + c}/{split_num ** 2}")
             masks = mask_generator.generate(small_image)
             num_masks = len(masks)
@@ -144,7 +141,7 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
     
     progress(0, desc="Generating Plot")
     # Create a figure with a size of 10 inches by 10 inches
-    fig = plt.figure(figsize=(40, 40))
+    fig = plt.figure(figsize=(10, 10))
 
     # Display the image using the imshow() function
     # plt.imshow(cropped_image)