Use osam-sam and osam-efficientsam

wkentaro · Jun 22, 2024 · 17e0b17 · 17e0b17
1 parent adaf2f7
commit 17e0b17
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 278 deletions.
diff --git a/osam/_models.py b/osam/_models.py
@@ -0,0 +1,3 @@
+# import modules to register the models
+import osam_sam  # noqa: F401
+import osam_efficientsam  # noqa: F401
diff --git a/osam/_models/__init__.py b/osam/_models/__init__.py
diff --git a/osam/_models/_efficient_sam.py b/osam/_models/_efficient_sam.py
diff --git a/osam/_models/_sam.py b/osam/_models/_sam.py
diff --git a/pyproject.toml b/pyproject.toml
@@ -34,6 +34,8 @@ dependencies = [
   "fastapi",
   "loguru",
   "osam-core @ git+https://github.com/wkentaro/osam-core.git@b54b13ea1c093980faaa03beccfc13853dadbb00",
+  "osam-efficientsam @ git+https://github.com/wkentaro/osam-efficientsam.git@46a8f397786d3c3235f16eafacc886e193c037cc",
+  "osam-sam @ git+https://github.com/wkentaro/osam-sam.git@3f12c6981269a52cce73552a21f87942457804de",
   "Pillow",
   "uvicorn",
 ]

diff --git a/src/.gitignore b/src/.gitignore
@@ -1 +1,3 @@
 /osam-core
+/osam-efficientsam
+/osam-sam