1. 首页
  2. 移动开发
  3. 其他
  4. TransBTS:此仓库提供了TransBTS的官方代码 源码

TransBTS:此仓库提供了TransBTS的官方代码 源码

上传者: 2021-03-30 11:10:36上传 ZIP文件 884.71KB 热度 64次
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
下载地址
用户评论