From e4cb92d3bde9eaaaab724594c317f7a8cad6230e Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Fri, 12 Apr 2024 18:01:06 +0200
Subject: [PATCH] Don't assume YXCZ stack when creating accessor
---
model_server/extensions/ilastik/models.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/model_server/extensions/ilastik/models.py b/model_server/extensions/ilastik/models.py
index be06d849..af280bcd 100644
--- a/model_server/extensions/ilastik/models.py
+++ b/model_server/extensions/ilastik/models.py
@@ -262,7 +262,7 @@ class IlastikObjectClassifierFromPixelPredictionsModel(IlastikModel, ImageToImag
raise InvalidInputImageError('Expecting input image and pixel probabilities to be the same shape')
pxch = kwargs.get('pixel_classification_channel', 0)
pxtr = kwargs.get('pixel_classification_threshold', 0.5)
- mask = InMemoryDataAccessor(pxmap.get_one_channel_data(pxch).data > pxtr)
+ mask = img._derived_accessor(pxmap.get_one_channel_data(pxch).data > pxtr)
obmap, _ = self.infer(img, mask)
return obmap
--
GitLab