Allow utilization of circular shape type for predicting polygons

healthonrails · Jan 11, 2024 · 3960be7 · 3960be7
1 parent 813f18b
commit 3960be7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/annolid/segmentation/SAM/edge_sam_bg.py b/annolid/segmentation/SAM/edge_sam_bg.py
@@ -244,7 +244,7 @@ def process_frame(self, frame_number):
                 continue
             if len(points) < 4:
                 orig_points = random_sample_near_center(
-                    Point(points), 4, 3)
+                    Point(points[0]), 4, 3)
             points = calculate_polygon_center(orig_points)
 
             polygon = Polygon(orig_points)