MedMLX — Segment. ENDOSCOPY / FLEXIBLEUNET

FlexibleUNet · Endoscopy

Segment surgical tools in RGB endoscopy frames with an EfficientNet-B2 FlexibleUNet.

Input	Output
RGB endoscopy frame	Full-resolution tool mask

Apple Silicon · Metal · Python 3.13 / 3.14

Quick start

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

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

hf download MedMLX/flexible-unet-effnetb2-mlx --local-dir models

Replace the example input paths with your own files.

from flexible_unet_mlx.medmlx import run

result = run(
    image_path='frame.jpg',
    weights_path='models/flexibleunet.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/flexible-unet-effnetb2-mlx

Base model

MONAI/endoscopic_tool_segmentation

Finetuned

(1)

this model