MedMLX — Segment. ABDOMINAL CT / SWINUNETR

SwinUNETR · Abdominal CT

Segment abdominal organs in 3D CT with the MONAI BTCV checkpoint and native MLX.

Input	Output
3D abdominal CT · NIfTI	13 organ labels + background

Apple Silicon · Metal · Python 3.13 / 3.14

Quick start

uv venv --python 3.13
source .venv/bin/activate

repo="https://github.com/MedMLX/swin-unetr-mlx"
release="$repo/releases/download/v0.2.1"
uv pip install "$release/swin_unetr_mlx-0.2.1-py3-none-any.whl"
uv tool install huggingface-hub

hf download MedMLX/swin-unetr-mlx --local-dir models

Replace the example input paths with your own files.

from swin_unetr_mlx.medmlx import run

result = run(
    image_path='ct.nii.gz',
    weights_path='models/swinunetr.safetensors',
    output_dir='outputs',
)
print(result["outputs"])

Use & license

For research and education. Not validated for clinical use.

Weights: Apache-2.0. Notices.

Downloads last month: -; Downloads are not tracked for this model. How to track

MLX

Hardware compatibility

Quantized

Model tree for MedMLX/swin-unetr-mlx

Base model

MONAI/swin_unetr_btcv_segmentation

Finetuned

(1)

this model