模型 | 路径 |
---|---|
U-net | https://github.com/bubbliiiing/unet-pytorch |
Unet 3+ | https://github.com/ZJUGiveLab/UNet-Version |
swin transformer | https://github.com/SwinTransformer/Swin-Transformer-Semantic-Segmentation |
Install the requirements.txt dependencies
pip install -r requirements.txt
Dataset preparation
https://www.kaggle.com/datasets/awsaf49/brats20-dataset-training-validation
https://www.kaggle.com/datasets/dschettler8845/brats-2021-task1
Running the data_progress.py
python data_progress.py
Running the train.py
CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python3 -m torch.distributed.launch --nproc_per_node 8 --master_port 20003 train.py
This will be presented after the paper is published.