Skip to content

Commit b840cb4

Browse files
Update elf dependency
1 parent a7dd369 commit b840cb4

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

environment_cpu.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ dependencies:
77
- cpuonly
88
- napari
99
- pooch
10-
- python-elf
10+
- python-elf >=0.4.8
1111
- pytorch
1212
- torchvision
1313
- tqdm

environment_gpu.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ name:
77
dependencies:
88
- napari
99
- pooch
10-
- python-elf
10+
- python-elf >=0.4.8
1111
- pytorch
1212
- pytorch-cuda>=11.7 # you may need to update the cuda version to match your system
1313
- torchvision

test/test_instance_segmentation.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)