Segmentation of fibroglandular tissue in breast MRI with neural networks

ArXiv: Fibroglandular Tissue Segmentation in Breast MRI using Vision Transformers - A multi-institutional evaluation

Figure: SwinTransformer for Breast Segmentation (TraBS)

• Segmentation masks for the DUKE dataset are saved at : data/Duke

Create virtual environment and install packages:
python -m venv venv
source venv/bin/activate
pip install -e .

• Go to breaststudies/data/dataset_breast.py and create a new Dataset based on BaseDataset.
• Go to breaststudies/data/datamodule_breast.py and create a new DataModule based on BaseDataModule.

• Optional: Pretrain model: scripts/main_pretrain.py
• Train model: scripts/main_train.py

• For the cross-fold prediction: scripts/main_predict_kfold.py
• For the (ensemble-) prediction on an external dataset: scripts/main_predict.py