Update helpers.py

autodistill · Aug 7, 2024 · d6eebed · d6eebed
1 parent 8fc5014
commit d6eebed
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/autodistill_grounded_sam_2/helpers.py b/autodistill_grounded_sam_2/helpers.py
@@ -70,6 +70,19 @@ def load_SAM():
 
     os.chdir(SAM_CACHE_DIR)
 
+    if not os.path.isdir("~/.cache/autodistill/segment_anything_2/segment-anything-2"):
+        subprocess.run(
+            [
+                "git",
+                "clone",
+                "https://github.com/facebookresearch/segment-anything-2.git",
+            ]
+        )
+
+        os.chdir("segment-anything-2")
+
+        subprocess.run(["pip", "install", "-e", "."])
+
     sys.path.append("~/.cache/autodistill/segment_anything_2/segment-anything-2")
 
     # Download the file if it doesn't exist