From b99e0f7ee41fd076b7a8bdf9f80405f348a73aee Mon Sep 17 00:00:00 2001 From: bd8090 Date: Fri, 9 Aug 2024 02:41:29 +0200 Subject: [PATCH] update README --- README.md | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/README.md b/README.md index 24a1e42..45a3b7a 100644 --- a/README.md +++ b/README.md @@ -167,6 +167,26 @@ And we will automatically save the tracking visualization results in `OUTPUT_VID > [!WARNING] > We initialize the box prompts on the first frame of the input video. If you want to start from different frame, you can refine `ann_frame_idx` by yourself in our code. +### Grounded-SAM-2 Video Object Tracking with Continuous ID (with Grounding DINO) +Users can upload their own video files and specify custom text prompts for grounding and tracking using the Grounding DINO and SAM 2 frameworks. To do this, execute the script: + + +```python +grounded_sam2_tracking_demo_with_continuous_id.py +``` + +You can customize various parameters including: + +- `text`: The grounding text prompt. +- `video_dir`: Directory containing the video files. +- `output_dir`: Directory to save the processed output. +- `output_video_path`: Path for the output video. +- `step`: Frame stepping for processing. +- `box_threshold`: box threshold for groundingdino model +- `text_threshold`: text threshold for groundingdino model +Note: This method supports only the mask type of text prompt. + + ### Citation If you find this project helpful for your research, please consider citing the following BibTeX entry.