search

Gy920 / segment-anything-2-real-time

Run Segment Anything Model 2 on a live video stream
Apache License 2.0
185 stars 36 forks source link

Add support for mask prompt? #13

Open reesekneeland opened 1 month ago

reesekneeland commented 1 month ago

Hello! Great work in this repository, I find this to be a much more natural use case for SAM2.

The original SAM2 repo has support for using an existing mask as a prompt for video segmentation via the add_new_mask() method, could this be implemented in your real time version?

Gy920 commented 1 month ago

Thank you for your suggestion. I have added add_new_mask() method. The specific usage is described here.