added sam prompt segmentation widget; fixed opencv version

juglab · Jan 14, 2024 · 6760c87 · 6760c87
1 parent 2a366b4
commit 6760c87
Show file tree

Hide file tree

Showing 5 changed files with 416 additions and 30 deletions.
diff --git a/env.yml b/env.yml
@@ -23,5 +23,5 @@ dependencies:
   - pip:
     - numpy
     - matplotlib
-    - opencv-python
+    - opencv-python==4.8.0.74
     - timm
diff --git a/setup.cfg b/setup.cfg
@@ -28,16 +28,16 @@ classifiers =
 packages = find:
 install_requires =
     numpy
-    opencv-python
+    opencv-python==4.8.0.74
     scikit-learn
     scikit-image
     matplotlib
     napari
     magicgui
     qtpy
     h5py
-    pytorch
-    torchvision
+    ; pytorch
+    ; torchvision
     timm
 
 python_requires = >=3.8

diff --git a/src/napari_sam_labeling_tools/__init__.py b/src/napari_sam_labeling_tools/__init__.py
@@ -3,9 +3,11 @@
 from ._embedding_extractor_widget import EmbeddingExtractorWidget
 from ._sam_predictor_widget import SAMPredictorWidget
 from ._sam_rf_segmentation_widget import SAMRFSegmentationWidget
+from ._sam_prompt_segmentation_widget import SAMPromptSegmentationWidget
 
 __all__ = (
     "EmbeddingExtractorWidget",
     "SAMPredictorWidget",
     "SAMRFSegmentationWidget",
+    "SAMPromptSegmentationWidget"
 )