quoccuongle/Grounded-SAM-2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refine video

Browse Source
This commit is contained in:
rentainhe
2024-08-09 19:23:55 +08:00
parent 4e0285a97b
commit 91914958bd
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
grounded_sam2_tracking_demo_with_continuous_id.py
View File

@@ -68,7 +68,7 @@ frame_names.sort(key=lambda p: int(os.path.splitext(p)[0]))
# init video predictor state
inference_state = video_predictor.init_state(video_path=video_dir)
step = 25 # the step to sample frames for Grounding DINO predictor
step = 10 # the step to sample frames for Grounding DINO predictor
sam2_masks = MaskDictionatyModel()
PROMPT_TYPE_FOR_VIDEO = "mask" # box, mask or point