70 lines
2.8 KiB
YAML
70 lines
2.8 KiB
YAML
# Model registry for the SAM backend.
|
|
#
|
|
# Add/remove entries freely. Paths may be absolute or relative to this file.
|
|
# Checkpoints marked as missing on disk are still listed by /models but flagged
|
|
# `available: false` so the frontend can grey them out.
|
|
#
|
|
# `family` selects the adapter code-path. Currently supported:
|
|
# sam2 — Meta SAM 2 / SAM 2.1 (image predictor, bbox prompts)
|
|
# sam3 — Meta SAM 3 (image predictor with box-as-exemplar, or HF tracker)
|
|
#
|
|
# Per-entry fields:
|
|
# id — stable key used by /predict?model=<id>
|
|
# name — human-readable, shown in the Tauri model dropdown
|
|
# family — adapter to use
|
|
# config — family-specific (SAM2: hydra yaml path; SAM3: "image_predictor"
|
|
# or "tracker")
|
|
# checkpoint — file path (relative paths resolved against this yaml's
|
|
# directory) OR an opaque identifier when `remote: true`
|
|
# remote — optional; true = treat checkpoint as a HuggingFace model id
|
|
# (no on-disk existence check)
|
|
|
|
default: sam2.1_hiera_small
|
|
|
|
models:
|
|
- id: sam2.1_hiera_tiny
|
|
name: "SAM 2.1 · Hiera Tiny"
|
|
family: sam2
|
|
config: configs/sam2.1/sam2.1_hiera_t.yaml
|
|
checkpoint: ../sam2/sam2/checkpoints/sam2.1_hiera_tiny.pt
|
|
|
|
- id: sam2.1_hiera_small
|
|
name: "SAM 2.1 · Hiera Small"
|
|
family: sam2
|
|
config: configs/sam2.1/sam2.1_hiera_s.yaml
|
|
checkpoint: ../sam2/sam2/checkpoints/sam2.1_hiera_small.pt
|
|
|
|
- id: sam2.1_hiera_base_plus
|
|
name: "SAM 2.1 · Hiera Base+"
|
|
family: sam2
|
|
config: configs/sam2.1/sam2.1_hiera_b+.yaml
|
|
checkpoint: ../sam2/sam2/checkpoints/sam2.1_hiera_base_plus.pt
|
|
|
|
- id: sam2.1_hiera_large
|
|
name: "SAM 2.1 · Hiera Large"
|
|
family: sam2
|
|
config: configs/sam2.1/sam2.1_hiera_l.yaml
|
|
checkpoint: ../sam2/sam2/checkpoints/sam2.1_hiera_large.pt
|
|
|
|
# ---- SAM 3 -----------------------------------------------------------
|
|
# Local image-predictor path — uses the .pt weights in this repo. Box is
|
|
# treated as a visual exemplar; the adapter picks the mask whose predicted
|
|
# bbox best overlaps the user bbox (classic single-instance annotation).
|
|
- id: sam3
|
|
name: "SAM 3 · image predictor"
|
|
family: sam3
|
|
config: image_predictor
|
|
checkpoint: ../files/sam3_weights/sam3.pt
|
|
|
|
# Optional HuggingFace tracker path. Requires `pip install transformers`
|
|
# and pulls weights from HF hub on first use. Gives cleaner single-box
|
|
# → single-mask results than the exemplar path but needs a network round
|
|
# trip the first time. Disabled by leaving `remote: false` — flip to
|
|
# `remote: true` once your machine has transformers installed.
|
|
# - id: sam3_tracker
|
|
# name: "SAM 3 · tracker (HuggingFace)"
|
|
# family: sam3
|
|
# config: tracker
|
|
# checkpoint: facebook/sam3
|
|
# remote: true
|