Improved Engine

mbasheer04 · Jan 24, 2025 · 3b51acb · 3b51acb
1 parent 46d30f2
commit 3b51acb
Show file tree

Hide file tree

Showing 4 changed files with 586 additions and 1,434 deletions.
diff --git a/examples/sam-architecture.ipynb b/examples/sam-architecture.ipynb
diff --git a/tests/models/test_general_segmentor.py b/tests/models/test_general_segmentor.py
diff --git a/tiatoolbox/models/__init__.py b/tiatoolbox/models/__init__.py
@@ -12,7 +12,7 @@
 from .architecture.sam import SAM
 from .engine.multi_task_segmentor import MultiTaskSegmentor
 from .engine.nucleus_instance_segmentor import NucleusInstanceSegmentor
-from .engine.sam_engine import GeneralSegmentor
+from .engine.general_segmentor import GeneralSegmentor
 from .engine.patch_predictor import (
     IOPatchPredictorConfig,
     PatchDataset,
@@ -35,7 +35,9 @@
     "MicroNet",
     "MultiTaskSegmentor",
     "NuClick",
+    "SAM",
     "NucleusInstanceSegmentor",
     "PatchPredictor",
     "SemanticSegmentor",
+    "GeneralSegmentor",
 ]
diff --git a/tiatoolbox/models/engine/sam_engine.py → ...oolbox/models/engine/general_segmentor.py b/tiatoolbox/models/engine/sam_engine.py → ...oolbox/models/engine/general_segmentor.py