add nms

app.py

@@ -172,7 +172,17 @@ def count_barnacles(image_raw, split_num, progress=gr.Progress()):
     
     
     # return annotated, mask_counter, centers
-    return fig, mask_counter, centers
+
+    good_centers = []
+    for point in centers:
+        is_good = True
+        for prev_point in good_centers:
+            if (point[0] - prev_point[0]) ** 2 + (point[1] + prev_point[1]) ** 2 < 20:
+                is_good = False
+        if is_good:
+            good_centers.append(point)
+    
+    return fig, mask_counter, good_centers
 
 demo = gr.Interface(count_barnacles,
                     inputs=[