AURORA
Segmentation models for cancer metastasis in brain MR
Install / Use
/learn @BrainLesion/AURORAREADME
AURORA
<!-- [](https://codecov.io/gh/BrainLesion/BraTS) -->Deep learning models for brain cancer metastasis segmentation based on the manuscripts:
- Identifying core MRI sequences for reliable automatic brain metastasis segmentation
- Development and external validation of an MRI-based neural network for brain metastasis segmentation in the AURORA multicenter study
Installation
With a Python 3.8+ environment, you can install brainles_aurora directly from pypi.org:
pip install brainles-aurora
Recommended Environment
- CUDA 11.4+ (https://developer.nvidia.com/cuda-toolkit)
- Python 3.8+
- GPU with CUDA support and at least 6GB of VRAM
Usage
BrainLes features Jupyter Notebook tutorials with usage instructions.
A minimal example could look like this:
from brainles_aurora.inferer import AuroraInferer, AuroraInfererConfig
config = AuroraInfererConfig(
tta=False, cuda_devices="4"
) # disable tta for faster inference in this showcase
inferer = AuroraInferer(config=config)
inferer.infer(
t1="t1.nii.gz",
t1c="t1c.nii.gz",
t2="t2.nii.gz",
fla="fla.nii.gz",
segmentation_file="segmentation.nii.gz",
whole_tumor_unbinarized_floats_file="whole_network.nii.gz",
metastasis_unbinarized_floats_file="metastasis_network.nii.gz",
log_file="aurora.log",
)
[!NOTE]
If you're interested in the AURORA package, the Brain Metastases Segmentation may also be of interest.
Citation
Please support our development by citing the following manuscripts:
Identifying core MRI sequences for reliable automatic brain metastasis segmentation
@article{buchner2023identifying,
title={Identifying core MRI sequences for reliable automatic brain metastasis segmentation},
author={Buchner, Josef A and Peeken, Jan C and Etzel, Lucas and Ezhov, Ivan and Mayinger, Michael and Christ, Sebastian M and Brunner, Thomas B and Wittig, Andrea and Menze, Bjoern H and Zimmer, Claus and others},
journal={Radiotherapy and Oncology},
volume={188},
pages={109901},
year={2023},
publisher={Elsevier}
}
also consider citing the original AURORA manuscript, especially when using the vanilla model (all 4 modalities as input):
@article{buchner2022development,
title={Development and external validation of an MRI-based neural network for brain metastasis segmentation in the AURORA multicenter study},
author={Buchner, Josef A and Kofler, Florian and Etzel, Lucas and Mayinger, Michael and Christ, Sebastian M and Brunner, Thomas B and Wittig, Andrea and Menze, Bj{\"o}rn and Zimmer, Claus and Meyer, Bernhard and others},
journal={Radiotherapy and Oncology},
year={2022},
publisher={Elsevier}
}
Contact / Feedback / Questions
If possible please open a GitHub issue here.
For inquiries not suitable for GitHub issues:
Florian Kofler florian.kofler [at] tum.de
Josef Buchner j.buchner [at] tum.de
Related Skills
node-connect
349.2kDiagnose OpenClaw node connection and pairing failures for Android, iOS, and macOS companion apps
frontend-design
109.5kCreate distinctive, production-grade frontend interfaces with high design quality. Use this skill when the user asks to build web components, pages, or applications. Generates creative, polished code that avoids generic AI aesthetics.
openai-whisper-api
349.2kTranscribe audio via OpenAI Audio Transcriptions API (Whisper).
qqbot-media
349.2kQQBot 富媒体收发能力。使用 <qqmedia> 标签,系统根据文件扩展名自动识别类型(图片/语音/视频/文件)。
