From 5bd8917877d720c4baaaec8202a31dcdbafd713a Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Wed, 20 Dec 2023 10:27:12 +0100
Subject: [PATCH] Added superclass for semantic segmentation models and
 modified ilastik pixel classification to inherit from it

---
 extensions/chaeo/examples/export_patch_focus_metrics.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/extensions/chaeo/examples/export_patch_focus_metrics.py b/extensions/chaeo/examples/export_patch_focus_metrics.py
index 8e5fa332..9e668a99 100644
--- a/extensions/chaeo/examples/export_patch_focus_metrics.py
+++ b/extensions/chaeo/examples/export_patch_focus_metrics.py
@@ -9,7 +9,7 @@ from skimage.filters import gaussian, sobel
 
 from extensions.ilastik.models import IlastikPixelClassifierModel
 from extensions.chaeo.products import export_3d_patches_with_focus_metrics, export_patches_from_zstack
-from extensions.chaeo.zmask import ZMaskObjectTable
+from extensions.chaeo.zmask import RoiSet
 from model_server.accessors import generate_file_accessor, InMemoryDataAccessor, write_accessor_data_to_file
 from model_server.workflows import Timer
 
@@ -58,7 +58,7 @@ def export_patch_focus_metrics_from_multichannel_zstack(
     #     filters=zmask_filters,
     #     expand_box_by=zmask_expand_box_by,
     # )
-    obj_table = ZMaskObjectTable(
+    obj_table = RoiSet(
         obmask.get_one_channel_data(pixel_class),
         stack.get_one_channel_data(zmask_channel),
         mask_type=mask_type,
-- 
GitLab