From 9a811694362f95c14baa90315db2114a9e1d255a Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Fri, 26 Apr 2024 14:55:35 +0200
Subject: [PATCH] Confirms that ilastik pixel object already works with
multichannel input
---
model_server/extensions/ilastik/tests/test_ilastik.py | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/model_server/extensions/ilastik/tests/test_ilastik.py b/model_server/extensions/ilastik/tests/test_ilastik.py
index 02195648..d98a41eb 100644
--- a/model_server/extensions/ilastik/tests/test_ilastik.py
+++ b/model_server/extensions/ilastik/tests/test_ilastik.py
@@ -344,6 +344,15 @@ class TestIlastikOnMultichannelInputs(TestServerBaseClass):
mod = ilm.IlastikPixelClassifierModel(params={'project_file': self.pa_px_classifier})
pxmap = mod.infer(self.input_image)[0]
self.assertEqual(pxmap.hw, self.input_image.hw)
+ self.assertEqual(pxmap.nz, self.input_image.nz)
+ return pxmap
+
+ def test_classify_objects(self):
+ pxmap = self.test_classify_pixels()
+ mod = ilm.IlastikObjectClassifierFromPixelPredictionsModel(params={'project_file': self.pa_ob_classifier})
+ obmap = mod.infer(self.input_image, pxmap)[0]
+ self.assertEqual(obmap.hw, self.input_image.hw)
+ self.assertEqual(obmap.nz, self.input_image.nz)
--
GitLab