[add] add cell_seg func

STOmics · Nov 4, 2023 · ed6567b · ed6567b
1 parent c355846
commit ed6567b
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/stereo/image/segmentation/seg_utils/v1_pro/__init__.py b/stereo/image/segmentation/seg_utils/v1_pro/__init__.py
@@ -1 +1 @@
-from stereo.image.segmentation.seg_utils.v1_pro.cell_seg_pipeline_v2 import CellSegPipeV1Pro  # noqa
+from stereo.image.segmentation.seg_utils.v1_pro.cell_seg_pipeline_v1_pro import CellSegPipeV1Pro  # noqa
diff --git a/stereo/image/segmentation/segment.py b/stereo/image/segmentation/segment.py
@@ -70,8 +70,6 @@ def cell_seg(
     os.environ["CUDA_VISIBLE_DEVICES"] = str(gpu)
     if version in [VersionType.v1.value, VersionType.v3.value] and not model_path:
         raise Exception("cell_seg() missing 1 required keyword argument: 'model_path'")
-    import pdb
-    pdb.set_trace()
     if version == VersionType.v1.value:
         cell_seg_pipeline = CellSegPipeV1(
             img_path,
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		from stereo.image.segmentation.seg_utils.v1_pro.cell_seg_pipeline_v2 import CellSegPipeV1Pro # noqa
		from stereo.image.segmentation.seg_utils.v1_pro.cell_seg_pipeline_v1_pro import CellSegPipeV1Pro # noqa