Commit Graph

7 Commits

Author SHA1 Message Date
Ronghang Hu
1034ee2a1a better support for non-CUDA devices (CPU, MPS) (#192) 2024-08-12 10:46:50 -07:00
Ronghang Hu
6ecb5ff8d0 Add interface for box prompt in SAM 2 video predictor (#174)
This PR adds an example to provide box prompt in SAM 2 as inputs to the `add_new_points_or_box` API (renamed from`add_new_points`, which is kept for backward compatibility). If `box` is provided, we add it as the first two points with labels 2 and 3, along with the user-provided points (consistent with how SAM 2 is trained).

The video predictor notebook `notebooks/video_predictor_example.ipynb` is updated to include segmenting from box prompt as an example.
2024-08-07 11:54:30 -07:00
Niels
43c385c263 Update docstrings 2024-08-06 23:00:26 +02:00
Niels
8f15c6255a Format using ufmt 2024-08-06 22:43:35 +02:00
Niels
841cc1f015 Update docstring 2024-08-05 09:44:03 +02:00
Niels
6aeee34775 Make huggingface_hub soft dependency 2024-08-05 09:37:53 +02:00
Haitham Khedr
0c5f8c5432 Initial commit 2024-07-29 21:54:20 +00:00