上传者: 42139042
|
上传时间: 2021-11-19 11:23:58
|
文件大小: 885KB
|
文件类型: -
TransBTS:使用变压器的多模式脑肿瘤分割
此是的正式实现。 可以从获取多模式脑肿瘤数据集(BraTS 2019)。
转站
3D TransBTS的体系结构。
要求
的Python 3.7
pytorch 1.6.0
火炬视觉0.7.0
泡菜
尼巴贝
数据预处理
从下载数据集后,需要进行数据预处理,这是将.nii文件转换为.pkl文件并实现日期归一化。
python3 preprocess.py
训练
在BraTS数据集上运行训练脚本。 分布式培训可用于培训建议的TransBTS,其中--nproc_per_node决定gpus的数量,而--master_port则暗示端口号。
python3 -m torch.distributed.launch --nproc_per_node=4 --master_port 20003 train.py
测验
如果要测试已在B