
TransBTS官方代码库:此仓库包含TransBTS的官方代码
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
简介:TransBTS官方代码库汇集了该模型的全部官方源码,便于研究人员和开发者学习、实验及进一步开发。
TransBTS:使用变压器的多模式脑肿瘤分割
这是正式实现的一部分。可以从相关来源获取多模式脑肿瘤数据集(BraTS 2019)。
3D TransBTS架构描述如下:
要求:
- Python 3.7
- PyTorch 1.6.0
- TorchVision 0.7.0
数据预处理步骤包括从下载的数据集中将.nii文件转换为.pkl文件,并进行日期归一化。
运行命令:`python3 preprocess.py`
训练过程涉及在BraTS数据集上使用训练脚本。分布式培训可用于建议的TransBTS模型,其中--nproc_per_node参数指定了使用的GPU数量,而--master_port则用于指定端口号。
运行命令:`python3 -m torch.distributed.launch --nproc_per_node=4 --master_port 20003 train.py`
测试已训练好的模型时,请使用相应的测试脚本。
全部评论 (0)
还没有任何评论哟~


