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
README.md
View File

@@ -197,7 +197,7 @@ Note: This method supports only the mask type of text prompt.
After running our demo code, you can get the tracking results as follows: After running our demo code, you can get the tracking results as follows:
[![Video Name](./assets/tracking_car_mask_1.jpg)](https://github.com/user-attachments/assets/d4b80e9b-bab3-4562-8b1a-9bf41e454b68) [![Video Name](./assets/tracking_car_mask_1.jpg)](https://github.com/user-attachments/assets/d3f91ad0-3d32-43c4-a0dc-0bed661415f4)
### Citation ### Citation

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 # init video predictor state
inference_state = video_predictor.init_state(video_path=video_dir) 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() sam2_masks = MaskDictionatyModel()
PROMPT_TYPE_FOR_VIDEO = "mask" # box, mask or point PROMPT_TYPE_FOR_VIDEO = "mask" # box, mask or point