Skip to content

Latest commit

 

History

History
39 lines (26 loc) · 971 Bytes

README.md

File metadata and controls

39 lines (26 loc) · 971 Bytes

Segment Anything

An implementation of the Segment Anything Model (SAM) in MLX. See the original repo by Meta AI for more details.1

Installation

pip install -r requirements.txt

Convert

python convert.py --hf-path facebook/sam-vit-base --mlx-path sam-vit-base

The safetensors weight file and configs are downloaded from Hugging Face, converted, and saved in the directory specified by --mlx-path.

The model sizes are:

  • facebook/sam-vit-base
  • facebook/sam-vit-large
  • facebook/sam-vit-huge

Run

See examples notebooks/predictor_example.ipynb and notebooks/automatic_mask_generator_example.ipynb to try the Segment Anything Model with MLX.

You can also generate masks from the command line:

python main.py --model <path/to/model> --input <image_or_folder> --output <path/to/output>

Footnotes

  1. The original Segment Anything GitHub repo. ↩