Skip to main content

A napari plugin for interactive Segment Anything Model 3 (SAM3) segmentation using text, points, boxes, exemplar prompts, large-image ROI inference, mask operations, and 3D/video-like propagation

Project description

napari-sam3-assistant

napari-sam3-assistant UI

napari-sam3-assistant is a napari plugin for interactive Segment Anything Model 3 (SAM3) segmentation using text, points, boxes, exemplar prompts, large-image ROI inference, point-based refinement, mask operations, and 3D/video-like propagation.

The plugin focuses on task-based segmentation workflows:

  • 2D segmentation with text, box, point, and mask-style prompts
  • 3D stack / video-like propagation from prompts on a selected slice or frame
  • exemplar segmentation from Shapes ROI boxes
  • text-based concept segmentation
  • large OME-Zarr and TIFF segmentation through local ROI inference
  • refinement with positive and negative point prompts
  • downstream mask cleanup, merge, and export operations

SAM 3 is not bundled with this plugin. Install the SAM 3 backend and download the SAM 3 model files separately from Meta's Hugging Face repository.

Status

This project is under active development. The current widget supports local SAM 3 model loading, napari prompt collection, large-image ROI execution, downstream mask operations, background execution, and writing results back to napari layers.

Changelog

Release notes and bug-fix history are maintained in CHANGELOG.md.

Requirements

  • Python >=3.12
  • napari >=0.5
  • SAM 3 Python package importable as sam3
  • PyTorch and torchvision installed for your platform
  • A local SAM 3 checkpoint directory containing:
    • config.json
    • processor_config.json
    • one weight file such as sam3.pt, model.safetensors, or sam3.1_multiplex.pt

If CUDA is not available or not compatible, select CPU in the widget.

Setup

If you installed the standalone napari desktop app from the website, Plugin Manager may not work for this plugin because that app uses a different Python environment.

Use a new Conda environment with Python 3.12 and install napari, SAM3, and napari-sam3-assistant there.

Windows

# 1) install Miniforge first from:
# https://conda-forge.org/download/

# 2) open Miniforge Prompt or PowerShell after Miniforge is installed


conda create -n napari-sam3 python=3.12 -y
conda activate napari-sam3

python -m pip install --upgrade pip wheel
python -m pip install "setuptools<82"
python -m pip install "napari[all]"

# choose one:
# CPU
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# GPU
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

Install SAM3 (Windows)

git clone https://github.com/facebookresearch/sam3.git
or
# quick alternative
python -m pip install --no-cache-dir git+https://github.com/facebookresearch/sam3


python -m pip install einops triton-windows pycocotools

Linux ARM64 (AArch64)

# 1) install Miniforge first from:
# https://conda-forge.org/download/

conda create -n napari-sam3 python=3.12 -y
conda activate napari-sam3

python -m pip install --upgrade pip wheel
python -m pip install "setuptools<82" "numpy>=1.26,<2"
pip install "napari[bermuda, pyqt6, optional-numba, optional-base]"

# choose one:
# CPU

python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu

# GPU

python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu130

Install SAM3 (Linux ARM64)

git clone https://github.com/facebookresearch/sam3.git
cd sam3
python -m pip install --no-cache-dir -e .

Download SAM 3 model files

Download the model files from:

  • https://huggingface.co/facebook/sam3
  • https://huggingface.co/facebook/sam3.1

These repositories are gated, so you must request or accept access first.

After approval, download the files manually from the Files and versions tab.

A reference screenshot of the file list is shown below:

SAM 3 model files screenshot

Model folder

The model files can be stored in any folder you want.

Examples:

D:\models\sam3
D:\models\sam3_1

The plugin only needs the correct folder path.

Example SAM3.0 folder:

D:\models\sam3\
  config.json
  processor_config.json
  sam3.pt

model.safetensors is also supported as a SAM3 3.0 weight file.

Example SAM3.1 folder:

D:\models\sam3_1\
  config.json
  processor_config.json
  sam3.1_multiplex.pt

Current model support:

  • SAM3.0 weights: 2D image tasks and 3D/video propagation
  • SAM3.1 sam3.1_multiplex.pt: 3D/video propagation through the SAM3.1 multiplex video predictor
  • SAM3.1 is not currently routed through the plugin's 2D image model loader

Device rule

  • If you installed CPU-only PyTorch, choose CPU
  • If you installed CUDA PyTorch and torch.cuda.is_available() is True, you may choose CUDA

For developers

If you want editable installs and git pull, use:

git clone https://github.com/facebookresearch/sam3.git
cd sam3
python -m pip install --no-cache-dir -e .

git clone https://github.com/wulinteousa2-hash/napari-sam3-assistant.git
cd napari-sam3-assistant
python -m pip install -e .

Basic Workflow

  1. Open an image in napari.
  2. Open Plugins > SAM3 Assistant.
  3. Select the image in Layers > Image.
  4. Select a task.
  5. Create a prompt layer if the task needs one.
  6. Click Run Preview.
  7. Inspect SAM3 preview labels, SAM3 preview masks, or SAM3 preview boxes.
  8. Click Save Result as Labels to keep the result.

Use Clear Preview to remove generated preview layers without deleting prompts or saved labels.

Large-Image Local Inference

Large-image mode is optional and off by default. When it is off, the plugin keeps the existing full-image inference path.

Use this mode for OME-Zarr, large TIFF, and similar large images where sending the full selected plane to SAM3 is too expensive.

Workflow:

  1. Set up the normal task and prompt type.
  2. Enable Enable large-image local inference in Step 2. Task.
  3. Choose a local ROI size:
512 x 512
1024 x 1024
2048 x 2048
4096 x 4096
8192 x 8192
  1. Add a point or box prompt.
  2. Click Run Preview, or add points in live refinement mode.

ROI behavior:

  • Point prompts use the latest point as the ROI anchor.
  • Box prompts use the box center and keep the box inside the local inference window when possible.
  • Refinement uses the latest point as the ROI anchor.
  • Text-only prompts keep the full-image path in this first pass unless a point or box anchor is also available.
  • If a new point or box stays inside the current ROI, the same ROI is reused.
  • If a new point or box falls outside the current ROI, the ROI is rebuilt around the new prompt.

The active ROI is shown as:

SAM3 active ROI

SAM3 receives only the local ROI image data. Returned labels and boxes are written back into global image coordinates in the normal preview layers.

Status messages report:

Large-image mode OFF: full-image inference.
Large-image mode ON: local ROI inference (WIDTH x HEIGHT).
Active ROI bounds: y=Y0:Y1, x=X0:X1.

Batch Multiple 2D Images

Use Batch all image layers when several open 2D images should receive the same prompt setup.

Workflow:

  1. Open multiple images in napari.
  2. Configure a 2D task such as text, box, exemplar, or labels-mask segmentation.
  3. Add the prompt once.
  4. Enable Batch all image layers in Step 3. Layers.
  5. Click Run Preview.

Each source image gets its own output layers:

SAM3 preview labels [image name]
SAM3 preview masks [image name]
SAM3 preview boxes [image name]

Save Result as Labels saves each batch preview labels layer separately:

SAM3 saved labels [image name]

Batch mode is intended for 2D image tasks. It is disabled for live refinement and 3D/video propagation because those workflows depend on one active image/session.

Multi-Text Batch Mode

Use Batch text prompts when you want each text concept to run independently instead of writing one combined phrase such as cat and dog.

Workflow:

  1. Set Task to Text segmentation.
  2. Enter one concept per line in Batch text prompts:
cat
dog
person
  1. Leave Batch all image layers off to run all prompts on the selected image only.
  2. Enable Batch all image layers to run every prompt on every open image.
  3. Click Run Preview.

Outputs include both image and prompt:

SAM3 preview labels [Image 1 - cat]
SAM3 preview labels [Image 1 - dog]
SAM3 preview labels [Image 2 - cat]
SAM3 preview labels [Image 2 - dog]

The Results table includes a Prompt column. Object IDs are scoped to each image-prompt result, so Object ID 1 for Image 1 - cat is separate from Object ID 1 for Image 1 - dog.

Tasks

Text Segmentation

Use text to segment all matching instances of a concept.

Workflow:

  1. Set Task to Text segmentation.
  2. Leave Prompt type as Text only.
  3. Enter a short phrase, for example:
cell
nucleus
myelin
myelin sheath
  1. Keep Detection threshold near the default 0.35, or lower it if the result is empty.
  2. Press Enter in the text prompt field or click Run Preview.

No prompt layer is needed for text segmentation. Create Prompt Layer is not required.

Text prompts usually work better as short noun phrases than instructions. Prefer myelin sheath over segment all the myelin rings. The plugin strips common instruction prefixes before sending the prompt to SAM3, but microscopy-specific language can still be difficult for the model.

If the result says objects=0, SAM3 ran but did not return masks above threshold. Try a shorter noun phrase, lower Detection threshold, or use a box/exemplar prompt for structures that are visually clear but not well recognized by text.

2D Segmentation With Boxes

Use boxes to identify the target object or concept.

Workflow:

  1. Set Task to 2D segmentation.
  2. Set Prompt type to Box.
  3. Click Create Prompt Layer.
  4. Draw one or more rectangles in the SAM3 boxes Shapes layer.
  5. Click Run Preview.

The output appears in preview layers.

Exemplar Segmentation

Use example ROIs to segment similar objects.

Workflow:

  1. Set Task to Exemplar segmentation.
  2. Set Prompt type to Box.
  3. Click Create Prompt Layer.
  4. Draw boxes around one or more example objects.
  5. Click Run Preview.

The local SAM 3 image API exposes visual exemplars through geometric box prompts. The plugin stores ROI metadata, but inference currently passes exemplar ROIs as SAM 3 visual box prompts.

Exemplar segmentation is a 2D/image task in this plugin. For exemplar-like 3D propagation, use 3D/video propagation with a box prompt on the selected frame or slice.

Refinement With Positive and Negative Points

Use points to correct a result.

Workflow:

  1. Set Task to Refinement (live point correction).
  2. Set Prompt type to Points (positive/negative).
  3. Click Create Prompt Layer.
  4. Choose Positive and add points on regions to include.
  5. Choose Negative and add points on regions to exclude.
  6. Click Run Preview.

This is useful after a text, box, or exemplar preview is close but not correct.

Labels Mask Prompt

Use a napari Labels layer as a mask-style prompt.

Workflow:

  1. Set a task that supports mask prompts.
  2. Set Prompt type to Labels mask.
  3. Click Create Prompt Layer.
  4. Paint non-zero pixels in SAM3 mask prompt.
  5. Click Run Preview.

Labels-mask prompts are currently supported for 2D/image workflows, not 3D/video propagation. For 3D/video propagation, use text, box, or point prompts on the selected frame.

3D Stack / Video Propagation

Treat a stack as video-like data and propagate a prompt through frames or slices.

Workflow:

  1. Open a stack in napari.
  2. Set Task to 3D/video propagation.
  3. Select the target frame or slice in napari.
  4. Create a prompt layer and add prompts on that frame.
  5. Choose propagation direction:
    • both
    • forward
    • backward
  6. Click Start 3D Propagation.

In 3D/video propagation mode, the primary run button changes from Run Preview to Start 3D Propagation. This starts a new SAM3 video session, adds the current frame prompt, and propagates through the stack. Propagate Existing Session is an advanced action that reuses the current SAM3 video session without adding a new prompt; it is enabled only after a successful 3D propagation run.

3D/video prompt limits:

  • SAM3.0 video propagation supports one initial visual box on the prompted frame.
  • SAM3.1 video multiplex can accept multiple box prompts.
  • Point prompts target one object per request and cannot be mixed with text or box prompts in the same 3D/video request.
  • Point prompts are limited to 16 points per request. SAM3's tracker prompt encoder uses the first 8 and last 8 points when more are supplied, so the plugin fails early instead of letting middle points be ignored.
  • Labels-mask prompts are not supported by the SAM3 video predictor API used by this plugin.
  • Exemplar segmentation itself is routed through the 2D image model. Use box prompts in 3D/video propagation when the goal is exemplar-like propagation through a stack.

Preview output is written to:

SAM3 propagated preview labels

Saved output is written to:

SAM3 saved propagated labels

The current SAM 3 video predictor backend is CUDA-only. CPU mode is supported for 2D/image workflows, not 3D/video propagation.

Channel Axis

Channel axis tells the plugin which data axis is color/channel.

Default:

-1

Use -1 for grayscale images and normal RGB/RGBA images. The plugin auto-detects trailing RGB/RGBA axes of size 3 or 4.

Examples:

(H, W)          -> -1
(H, W, 3)      -> -1
(H, W, 4)      -> -1
(Z, H, W)      -> -1
(C, H, W)      -> 0
(Z, C, H, W)   -> 1
(T, C, H, W)   -> 1
(Z, H, W, C)   -> 3

Leave it at -1 unless your image has an explicit multi-channel microscopy dimension.

Output Layers

Preview layers:

SAM3 preview labels
SAM3 preview masks
SAM3 preview boxes
SAM3 propagated preview labels

Saved layers:

SAM3 saved labels
SAM3 saved propagated labels

Buttons:

  • Validate: check the selected SAM 3 model directory.
  • Load 2D Model: load the 2D/image model.
  • Load 3D/Video Model: load the video propagation model.
  • Run Preview: run the selected task.
  • Clear Preview: remove generated preview layers only.
  • Save Result as Labels: copy preview labels into saved labels.
  • Cancel: stop a running worker.
  • Unload: unload the SAM3 model from memory.

Results table:

Layer | Prompt | Frame | Object ID | Score | Area
  • Layer: source image layer.
  • Prompt: text prompt used for text and multi-text results. For non-text workflows this is -.
  • Frame: propagated frame or slice index. For 2D results this is -.
  • Object ID: SAM3 object ID when available, otherwise a generated label ID.
  • Score: SAM3 confidence/probability when returned by the backend.
  • Area: number of mask pixels for that object in the displayed 2D plane or frame.

Results actions:

  • Clear Results: clear the table only.
  • Copy Clipboard: copy tab-separated results, including headers, for pasting into Excel or statistics software.
  • Export CSV: save the results table to a CSV file.

Label-value merge:

Use this when multiple SAM3 objects should become the same class value in a Labels layer. For example, if labels 3, 4, 5, and 6 are all the same biological class, set:

Values to replace: 3,4,5,6
New value: 3

Then click Merge Label Values. The selected Labels layer is updated in place.

Step 7 Mask Operations

Step 7 provides downstream mask operations for turning SAM3 previews into curated masks for analysis or training data.

Tabs:

  • Accepted Objects: save a preview Labels layer as a named accepted object with class metadata, append it to an existing accepted layer, or replace an existing accepted layer.
  • Class Merge: merge selected accepted-object layers into a class working mask.
  • Mask Cleanup: analyze connected components, delete selected components, remove small objects, fill holes, smooth masks, keep the largest component, and relabel values.
  • Final Merge / Export: merge cleaned class masks into semantic, instance, or binary final masks, choose overlap handling, and export TIFF, PNG, or NumPy .npy files.

The mask operations panel works on napari Labels layers, including SAM3 preview and saved label layers.

ARM64, CUDA, and DGX Spark

For ARM64 systems such as NVIDIA DGX Spark / GB10:

  • Use Python 3.12 or newer.
  • Keep the NVIDIA driver and CUDA stack current.
  • Install a PyTorch/torchvision build that supports your GPU architecture.
  • Use CPU mode for reliable 2D execution if CUDA kernels are unavailable.
  • Use explicit CUDA only when testing a compatible GPU build.

Check PyTorch GPU support:

python - <<'PY'
import torch
print("torch:", torch.__version__)
print("torch cuda runtime:", torch.version.cuda)
print("cuda available:", torch.cuda.is_available())
if torch.cuda.is_available():
    print("device:", torch.cuda.get_device_name(0))
    print("capability:", torch.cuda.get_device_capability(0))
    print("arch list:", torch.cuda.get_arch_list())
PY

GB10 reports compute capability 12.1 (sm_121). If your PyTorch build does not include compatible kernels, you may see:

CUDA error: no kernel image is available for execution on the device
nvrtc: error: invalid value for --gpu-architecture

The plugin does not compile PyTorch, torchvision, or SAM 3 CUDA extensions.

Troubleshooting

No mask appears and status says objects=0

SAM 3 returned no detections above threshold. Try:

  • a shorter text prompt
  • a more common concept phrase
  • a lower Detection threshold
  • box or exemplar prompts
  • CPU mode if the CUDA path is unstable

CUDA kernel image error

Error:

CUDA error: no kernel image is available for execution on the device

The GPU is visible, but at least one required CUDA kernel was not built for the device architecture. Use CPU, or install compatible PyTorch/torchvision/SAM 3 builds.

Invalid GPU architecture

Error:

nvrtc: error: invalid value for --gpu-architecture

The installed PyTorch CUDA runtime cannot compile for the detected GPU. Use CPU or install a build that supports the GPU.

BFloat16 conversion errors

The plugin converts SAM3 bfloat16 outputs to float32 before writing NumPy-backed napari layers. If you still see dtype errors, restart napari after changing device mode and run again.

Text prompt creates no layer

That is expected. Text segmentation does not need a prompt layer. Enter text and click Run Preview.

Development

Install in editable mode:

pip install -e .

Run tests:

PYTHONPATH=src pytest -q

The test suite covers coordinate mapping, prompt collection, adapter utility behavior, and static widget UI checks. It does not download SAM 3 weights.

References

Acknowledgement

The demo image was provided by the Electron Microscopy Core Facility at Houston Methodist Research Institute

License

MIT. See the project license file.

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

napari_sam3_assistant-3.3.0.tar.gz (75.4 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

napari_sam3_assistant-3.3.0-py3-none-any.whl (75.2 kB view details)

Uploaded Python 3

File details

Details for the file napari_sam3_assistant-3.3.0.tar.gz.

File metadata

  • Download URL: napari_sam3_assistant-3.3.0.tar.gz
  • Upload date:
  • Size: 75.4 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.12.13

File hashes

Hashes for napari_sam3_assistant-3.3.0.tar.gz
Algorithm Hash digest
SHA256 c4f61a219221fdc7860fd8aa25cb6064f3a5b191f75e89ce0908c8aabbdb61e2
MD5 5f9cc3deb9938d90154ccc3fe02b1110
BLAKE2b-256 309ac2516dde571a79054b272fb2c90eaa81743385f44b711f725fe58e878880

See more details on using hashes here.

File details

Details for the file napari_sam3_assistant-3.3.0-py3-none-any.whl.

File metadata

File hashes

Hashes for napari_sam3_assistant-3.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 34465212146eca987efa9b790df33f326ca9580c95efcacafe61af0bc17e8889
MD5 7254332e8a7b07eb03fa9de34bd53b54
BLAKE2b-256 499a05b66fd663a863e8628303bc61e5a56c3222ac27a1b7a19f2dd0e3d6dcc5

See more details on using hashes here.

Supported by

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