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/sam2/utils/misc.py
+++ b/sam2/utils/misc.py
@@ -184,7 +184,7 @@ def load_video_frames(
    frame_names = [
        p
        for p in os.listdir(jpg_folder)
-        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]))
    num_frames = len(frame_names)