模型 | 路径 |
---|---|
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 |
安装requirements.txt依赖
pip install -r requirements.txt
数据集准备
https://www.kaggle.com/datasets/awsaf49/brats20-dataset-training-validation
https://www.kaggle.com/datasets/dschettler8845/brats-2021-task1
运行data_progress.py
运行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
@article{xu2024brain,
title={Brain tumour segmentation framework with deep nuanced reasoning and Swin-T},
author={Xu, Yang and Yu, Kun and Qi, Guanqiu and Gong, Yifei and Qu, Xiaolong and Yin, Li and Yang, Pan},
journal={IET Image Processing},
year={2024},
publisher={Wiley Online Library}
}