Skip to main content

Metasam A Python package for Inference SAM

Project description

MetaSam: Packaged version of the Segment Anything 2 Model

teaser

🛠️ Installation

pip install metasam

🤗 Model Hub

bash script/download_model.sh

⭐ Usage

from metasam import SAM2Wrapper

# Initialize SAM2Wrapper
sam = SAM2Wrapper("path/to/checkpoint", "path/to/config")

# Load an image
sam.set_image("path/to/your/image.jpg")

# Predict segmentation
masks, scores, logits = sam.predict(point_coords=[[500, 640]], point_labels=[1])

# Visualize results
sam.show_masks(masks, scores)

😍 Contributing

pip install pre-commit
pre-commit install
pre-commit run --all-files

📜 License

This project is licensed under the terms of the Apache License 2.0.

🤗 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 arXiv:2408.00714},
  url={https://arxiv.org/abs/2408.00714},
  year={2024}
}

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

metasam-0.0.3.tar.gz (64.9 kB view details)

Uploaded Source

File details

Details for the file metasam-0.0.3.tar.gz.

File metadata

  • Download URL: metasam-0.0.3.tar.gz
  • Upload date:
  • Size: 64.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.10.12

File hashes

Hashes for metasam-0.0.3.tar.gz
Algorithm Hash digest
SHA256 2c91e985158fdf40439266c46579671545114b5243388c526bc4e08c2d874616
MD5 e1ad10311aa3639c238d2673f572fae1
BLAKE2b-256 a262c5878d8396a5e41cabfa1a863f865249af4b7e5dc245f59488e7b90498ae

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page