updated paths

.gitignore

@@ -2,5 +2,4 @@
 .DS_Store
 gradio_cached_examples/**
 *.png
-*.jpeg
-*.pth
+*.jpeg

app.py

@@ -6,8 +6,8 @@ os.environ["CUDA_HOME"] = "/usr/local/cuda-11.8/"
 subprocess.run(['pip', 'install', '-e', 'GroundingDINO'])
 sys.path.append(os.path.join(os.getcwd(), "GroundingDINO"))
 sys.path.append(os.path.join(os.getcwd(), "segment_anything"))
-os.system("wget https://huggingface.co/ShilongLiu/GroundingDINO/resolve/main/groundingdino_swint_ogc.pth")
-os.system("wget https://huggingface.co/spaces/mrtlive/segment-anything-model/resolve/main/sam_vit_h_4b8939.pth")
+# os.system("wget https://huggingface.co/ShilongLiu/GroundingDINO/resolve/main/groundingdino_swint_ogc.pth")
+# os.system("wget https://huggingface.co/spaces/mrtlive/segment-anything-model/resolve/main/sam_vit_h_4b8939.pth")
 
 import gradio as gr
 from dino_sam import sam_dino_vid

dino_sam.py

@@ -34,7 +34,7 @@ def sam_dino_vid(
         fps_processed: int = 1,
         video_options: list[str] = ["Bounding boxes"],
         config_path: str = "GroundingDINO/groundingdino/config/GroundingDINO_SwinT_OGC.py",
-        weights_path: str = "groundingdino_swint_ogc.pth",
+        weights_path: str = "weights/groundingdino_swint_ogc.pth",
         device: str = 'cuda',
         batch_size: int = 10
         ) -> (str, str): 
@@ -49,7 +49,7 @@ def sam_dino_vid(
         boxes_needed = False
     if "Masks" in video_options:
         masks_needed = True
-        checkpoint = "sam_vit_h_4b8939.pth"
+        checkpoint = "weights/sam_vit_h_4b8939.pth"
         model_type = "vit_h"
         sam = sam_model_registry[model_type](checkpoint=checkpoint)
         sam.to(device=device)
@@ -58,9 +58,9 @@ def sam_dino_vid(
     # create new dirs and paths for results
     filename = os.path.splitext(os.path.basename(vid_path))[0]
     results_dir = "../processed/" + filename +  datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
-    os.mkdir(results_dir)
+    os.makedirs(results_dir, exist_ok=True)
     frames_dir = os.path.join(results_dir, "frames")
-    os.mkdir(frames_dir)
+    os.makedirs(frames_dir, exist_ok=True)
     csv_path = os.path.join(results_dir, "detections.csv")
 
     # load the groundingDINO model

weights/groundingdino_swint_ogc.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b3ca2563c77c69f651d7bd133e97139c186df06231157a64c507099c52bc799
+size 693997677

weights/sam_vit_h_4b8939.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a7bf3b02f3ebf1267aba913ff637d9a2d5c33d3173bb679e46d9f338c26f262e
+size 2564550879