Environment

• Python 3.9
• Torch 1.12.1

Data Processing Steps

Data Preprocessing

  python preprocess.py

Train and Test

python train.py
python test.py

Model Weights

We have retrained our model on an NVIDIA RTX 4090. You can download it if needed.

Weight: Weight-BraTS2020

Log: Log-BraTS2020

Acknowledgments

We would like to extend our gratitude to the following project:

• D3Dnet