Segmentation of fibroglandular tissue in breast MRI with neural networks
Figure: SwinTransformer for Breast Segmentation (TraBS)
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