Skip to content

CPU compatible fork of the official SAMv2 implementation aimed at more accessible and documented tutorials

License

Notifications You must be signed in to change notification settings

SauravMaheshkar/samv2

Repository files navigation

Installation

This is a CPU compatible fork of the official SAMv2 implementation. You can download it from pypi using pip as follows:

pip install samv2

Usage

After downloading the official weights, you can use the load_model() helper method to instantiate a model.

from sam2 import load_model

model = load_model(
    variant="tiny",
    ckpt_path="artifacts/sam2_hiera_tiny.pt",
    device="cpu"
)

Features ๐Ÿš€

  • CPU compatible
  • ships with config files

Citation

@article{ravi2024sam2,
  title={SAM 2: Segment Anything in Images and Videos},
  author={Ravi, Nikhila and Gabeur, Valentin and Hu, Yuan-Ting and Hu, Ronghang and Ryali, Chaitanya and Ma, Tengyu and Khedr, Haitham and R{\"a}dle, Roman and Rolland, Chloe and Gustafson, Laura and Mintun, Eric and Pan, Junting and Alwala, Kalyan Vasudev and Carion, Nicolas and Wu, Chao-Yuan and Girshick, Ross and Doll{\'a}r, Piotr and Feichtenhofer, Christoph},
  journal={arXiv preprint},
  year={2024}
}