@@ -71,7 +71,7 @@ def test_automatic_mask_generator(self):
7171
7272 predicted = amg .generate ()
7373 predicted = mask_data_to_segmentation (predicted , image .shape , with_background = True )
74- self .assertGreater (matching (predicted , mask , threshold = 0.75 )["accuracy " ], 0.99 )
74+ self .assertGreater (matching (predicted , mask , threshold = 0.75 )["segmentation_accuracy " ], 0.99 )
7575
7676 # check that regenerating the segmentation works
7777 predicted2 = amg .generate ()
@@ -97,7 +97,7 @@ def test_embedding_mask_generator(self):
9797 predicted = amg .generate (pred_iou_thresh = 0.96 )
9898 predicted = mask_data_to_segmentation (predicted , image .shape , with_background = True )
9999
100- self .assertGreater (matching (predicted , mask , threshold = 0.75 )["accuracy " ], 0.99 )
100+ self .assertGreater (matching (predicted , mask , threshold = 0.75 )["segmentation_accuracy " ], 0.99 )
101101
102102 initial_seg = amg .get_initial_segmentation ()
103103 self .assertEqual (initial_seg .shape , image .shape )
@@ -126,7 +126,7 @@ def test_tiled_embedding_mask_generator(self):
126126 predicted = amg .generate (pred_iou_thresh = 0.96 )
127127 initial_seg = amg .get_initial_segmentation ()
128128
129- self .assertGreater (matching (predicted , mask , threshold = 0.75 )["accuracy " ], 0.99 )
129+ self .assertGreater (matching (predicted , mask , threshold = 0.75 )["segmentation_accuracy " ], 0.99 )
130130 self .assertEqual (initial_seg .shape , image .shape )
131131
132132 predicted2 = amg .generate (pred_iou_thresh = 0.96 )
@@ -151,7 +151,7 @@ def test_tiled_automatic_mask_generator(self):
151151 amg .initialize (image , image_embeddings = image_embeddings , verbose = False )
152152 predicted = amg .generate (pred_iou_thresh = pred_iou_thresh )
153153 predicted = mask_data_to_segmentation (predicted , image .shape , with_background = True )
154- self .assertGreater (matching (predicted , mask , threshold = 0.75 )["accuracy " ], 0.99 )
154+ self .assertGreater (matching (predicted , mask , threshold = 0.75 )["segmentation_accuracy " ], 0.99 )
155155
156156 predicted2 = amg .generate (pred_iou_thresh = pred_iou_thresh )
157157 predicted2 = mask_data_to_segmentation (predicted2 , image .shape , with_background = True )
0 commit comments