Metasam A Python package for Inference SAM
Project description
🛠️ 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
Release history Release notifications | RSS feed
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)
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
Algorithm | Hash digest | |
---|---|---|
SHA256 | 2c91e985158fdf40439266c46579671545114b5243388c526bc4e08c2d874616 |
|
MD5 | e1ad10311aa3639c238d2673f572fae1 |
|
BLAKE2b-256 | a262c5878d8396a5e41cabfa1a863f865249af4b7e5dc245f59488e7b90498ae |