Skip to main content

swin-unetr-mlx

Segment 13 abdominal organs in CT with MONAI SwinUNETR and native MLX.

Docs · MedMLX

Abdominal CT → multi-organ segmentation.

Abdominal CT → multi-organ segmentation.

Input Output
3D abdominal CT · NIfTI Organ mask on the input grid

Apple Silicon · Metal · Python 3.13 / 3.14 · uv

Quick start

uv venv --python 3.13
source .venv/bin/activate
uv pip install "swin-unetr-mlx[conversion]" huggingface-hub

Download and convert the checkpoint once, then run:

from pathlib import Path

from huggingface_hub import hf_hub_download
from swin_unetr_mlx import convert_checkpoint
from swin_unetr_mlx.medmlx import run

checkpoint = hf_hub_download("MONAI/swin_unetr_btcv_segmentation", "models/model.pt")
convert_checkpoint(Path(checkpoint), Path("model.safetensors"))

result = run(image_path="ct.nii.gz", weights_path="model.safetensors", output_dir="out")
print(result["outputs"]["mask"])

License

Apache-2.0. See third-party notices. Model weights retain their publisher terms.

For research and education. Outputs require human review.

Metadata

Release files for swin-unetr-mlx 0.2.2

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for swin-unetr-mlx 0.2.2
File Size Uploaded Provenance Actions
swin_unetr_mlx-0.2.2.tar.gz 27.3 kB Without provenance Details for swin_unetr_mlx-0.2.2.tar.gz

Built distribution (wheel)

Table of built distributions (wheels) for swin-unetr-mlx 0.2.2
File Interpreter ABI Platform Actions
swin_unetr_mlx-0.2.2-py3-none-any.whl Python 3 none any Details for swin_unetr_mlx-0.2.2-py3-none-any.whl

Total release size: 64.2 kB

Release files / swin_unetr_mlx-0.2.2.tar.gz

Download URL swin_unetr_mlx-0.2.2.tar.gz
Size 27.3 kB
Tags Source
SHA-256 checksum
How to use checksums
8ec0fe8a27af4ff421907d43501205cc564e7f1b4a96bb00a973d9f9eec53d8d
BLAKE2b-256 checksum
How to use checksums
f80ab8bb3b032da6cb793dca453a1d408bb4e49e2a97dc27c790b80ff869fb39
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.13.5

Release files / swin_unetr_mlx-0.2.2-py3-none-any.whl

Download URL swin_unetr_mlx-0.2.2-py3-none-any.whl
Size 36.9 kB
Tags Python 3
SHA-256 checksum
How to use checksums
3bcf86c6926c52015c5ad172b3399bef42ee52114ddb23df29de6374a7ad3195
BLAKE2b-256 checksum
How to use checksums
e25b8a0ad67a32dcf450762816a8a75cf99e6ec5240cd823fcb306e4cf93d3e7
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/7.0.0 CPython/3.13.5

Security for swin-unetr-mlx 0.2.2

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.2.2 This release

2 release files No provenance

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page