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

..

configs

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

csrc

fix: correct spelling

2024-07-30 08:13:47 -04:00

modeling

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

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

utils

Update links after renaming the repo from segment-anything-2 to sam2 (#341)

2024-09-30 20:27:44 -07:00

__init__.py

SAM2.1

2024-09-29 05:49:56 +00:00

automatic_mask_generator.py

accept kwargs in auto_mask_generator

2024-08-13 00:02:36 +00:00

benchmark.py

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

build_sam.py

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

sam2_hiera_b+.yaml

SAM2.1

2024-09-29 05:49:56 +00:00

sam2_hiera_l.yaml

SAM2.1

2024-09-29 05:49:56 +00:00

sam2_hiera_s.yaml

SAM2.1

2024-09-29 05:49:56 +00:00

sam2_hiera_t.yaml

SAM2.1

2024-09-29 05:49:56 +00:00

sam2_image_predictor.py

open README.md with unicode (to support Hugging Face emoji); fix various typos (#218)

2024-08-14 09:06:25 -07:00

sam2_video_predictor_legacy.py

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

sam2_video_predictor.py

patch for the case of offload_state_to_cpu=True in the new SAM2VideoPredictor (#490)

2024-12-12 15:12:13 -08:00