Grounded-SAM-2

Files

Ronghang Hu 2b90b9f5ce remove .pin_memory() in obj_pos of SAM2Base to resolve and error in MPS (#495 )

In this PR, we remove `.pin_memory()` in `obj_pos` of `SAM2Base` to resolve and error in MPS. Investigations show that `.pin_memory()` causes an error of `Attempted to set the storage of a tensor on device "cpu" to a storage on different device "mps:0"`, as originally reported in https://github.com/facebookresearch/sam2/issues/487.

(close https://github.com/facebookresearch/sam2/issues/487)

2024-12-15 16:47:17 -08:00

backbones

SAM 2 Update 12/11/2024 -- full model compilation for a major VOS speedup and a new SAM2VideoPredictor to better handle multi-object tracking (#486 )

2024-12-11 15:00:55 -08:00

sam

SAM 2 Update 12/11/2024 -- full model compilation for a major VOS speedup and a new SAM2VideoPredictor to better handle multi-object tracking (#486 )

2024-12-11 15:00:55 -08:00

__init__.py

Initial commit

2024-07-29 21:54:20 +00:00

memory_attention.py

Initial commit