Downloads · 30 days
0
Ricardooyarzun/brain-mets-autocontour-t1post
brain-mets-autocontour-t1post is a image segmentation model from Ricardooyarzun. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. It is set up for nnunet. The card lists the license as other.
Model card only — weights are not distributed here. See Weight availability.
Downloads · 30 days
0
Access
Public
Updated Aug 30, 2026
Repo size
—
Likes
0
Public
Click a slice to open those files.
.md4.6 KB · 75%
From the Hugging Face model README
Model card only — weights are not distributed here. See Weight availability.
A 3D nnU-Net trained to segment contrast-enhancing brain metastases on post-contrast T1 MRI, built as part of a working radiotherapy auto-contouring tool that exports DICOM RT Structure Sets directly importable into a treatment planning system.
Developed by a physician with three years of radiation oncology training, working as an ML engineer.
Research use only. This model is not a medical device, has not undergone independent clinical validation, and must not be used to inform patient care.
Intended for research on automated target delineation for brain metastasis radiotherapy planning. Any clinical use would require independent validation, regulatory clearance (MDR in the EU / FDA in the US), and a qualified clinician reviewing every contour.
| Architecture | nnU-Net v2, 3d_fullres |
| Input | Post-contrast T1 MRI (single channel), skull-stripped |
| Output | Binary segmentation of enhancing metastases (GTV) |
| Ensemble | 5 folds |
| Training | Custom trainer with early stopping |
| Framework | PyTorch + CUDA |
| Metric | Value | What it is |
|---|---|---|
| Dice (fold 0) | 0.8389 | nnU-Net internal EMA foreground pseudo-Dice, epoch 87 |
| Dice (5 folds, mean) | 0.8493 | range 0.8389 – 0.8642 |
| Inference latency | 0.84 s | median, full volume, 5-fold ensemble |
| Peak GPU memory | 2.07 GB | RTX 6000 Ada, PyTorch 2.11 + CUDA 12.8 |
Read the Dice figure correctly. It is nnU-Net's own held-out-fold metric computed during training, not independent validation against expert contours on unseen patients. No such independent evaluation has been performed, and no clinical Dice should be attributed to this model until it has.
Latency excludes DICOM→NIfTI conversion, skull stripping and RTSS export; end-to-end wall clock was not instrumented.
UCSF-BMSR — University of California San Francisco Brain Metastases Stereotactic Radiosurgery MRI Dataset, release 1.3. 560 brain MRIs from 412 patients with expert annotations of 5,136 metastases.
Rudie JD, et al. The University of California San Francisco Brain Metastases Stereotactic Radiosurgery (UCSF-BMSR) MRI Dataset. Radiology: Artificial Intelligence (2024). 10.1148/ryai.230126
Weights are deliberately not published here.
The UCSF-BMSR data use terms are custom and specify non-commercial, attribution, no redistribution, no re-identification. Whether publishing weights derived from that data constitutes redistribution is not settled by the licence text available to us, so the conservative reading applies: nothing derived from the dataset is redistributed until the data owners confirm otherwise.
This card documents the model so the work is inspectable and reproducible in description. It is not a workaround — no weights, no images, no data.
Researchers who want to evaluate the model can contact the author; a request for explicit permission from the dataset owners is the intended path to releasing weights.
Part of a portfolio of medical-imaging ML projects covering head-and-neck GTV segmentation benchmarking, breast BI-RADS assessment with external validation, and benchmark-contamination auditing.
Ricardo Oyarzún-Silva — physician (three years of radiation oncology training) and ML engineer. ORCID 0009-0000-3086-9047