From 8a46a99a5f6d556d235959e2da3b78366dc08ed0 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Mon, 23 Sep 2024 10:33:44 +0200
Subject: [PATCH] Call MIP method directly
---
model_server/base/roiset.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py
index 954e878d..9b270de5 100644
--- a/model_server/base/roiset.py
+++ b/model_server/base/roiset.py
@@ -91,10 +91,9 @@ def get_label_ids(acc_seg_mask: GenericImageDataAccessor, allow_3d=False, connec
la_3d[:, :, 0, zi] = la_2d
return InMemoryDataAccessor(la_3d)
else:
- # TODO: call argmax z method
return InMemoryDataAccessor(
label(
- acc_seg_mask.data_xyz.max(axis=-1),
+ acc_seg_mask.get_mip().data_xy,
connectivity=1,
).astype('uint16')
)
--
GitLab