文件名称：TransBTS:此仓库提供了TransBTS的官方代码
文件大小：885KB
文件格式：ZIP
更新时间：2024-04-01 11:51:08
Python
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
【文件预览】：
TransBTS-main
----train.py(10KB)
----utils()
--------__init__.py(2B)
--------tools.py(207B)
----models()
--------TransBTS()
--------criterions.py(5KB)
----test.py(4KB)
----predict.py(9KB)
----LICENSE(11KB)
----README.md(2KB)
----figure()
--------quantitive_comparison.PNG(198KB)
--------visual_comparison.PNG(446KB)
--------TransBTS.PNG(236KB)
----data()
--------preprocess.py(3KB)
--------BraTS.py(5KB)