feat: continuous_id_plus

2024-08-16 01:46:41 +02:00
parent 5f886743d9
commit 245fc7206b
8 changed files with 274 additions and 10 deletions
--- a/grounded_sam2_tracking_demo_with_gd1.5.py
+++ b/grounded_sam2_tracking_demo_with_gd1.5.py
@@ -44,7 +44,7 @@ video_dir = "notebooks/videos/bedroom"
 # scan all the JPEG frame names in this directory
 frame_names = [
    p for p in os.listdir(video_dir)
-    if os.path.splitext(p)[-1] in [".jpg", ".jpeg", ".JPG", ".JPEG"]
+    if os.path.splitext(p)[-1] in [".jpg", ".jpeg", ".JPG", ".JPEG", ".png", ".PNG"]
 ]
 frame_names.sort(key=lambda p: int(os.path.splitext(p)[0]))