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

refine README

Browse Source
This commit is contained in:
rentainhe
2024-08-09 09:36:03 +08:00
parent 0ba553bf88
commit 3a0089e7cb
4 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
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 = 10 # the step to sample frames for groundedDino predictor
step = 25 # the step to sample frames for Grounding DINO predictor
sam2_masks = MaskDictionatyModel()
PROMPT_TYPE_FOR_VIDEO = "mask" # box, mask or point
@@ -195,4 +195,4 @@ Step 6: Draw the results and save the video
"""
CommonUtils.draw_masks_and_box(video_dir, mask_data_dir, json_data_dir, result_dir)
create_video_from_images(result_dir, output_video_path, frame_rate=15)
create_video_from_images(result_dir, output_video_path, frame_rate=30)