You can find datasets here :
https://www.kaggle.com/datasets/awsaf49/brats20-dataset-training-validation
In the location BraTS2020_TrainingData/MICCAI_BraTS2020_TrainingData/
Look for BraTS20_Training_355 folder. The segmentation name inside the 355th folder is different. Change segmentation of it to BraTS20_Training_355_seg.nii
Usage :
pip install -r requirements.txt
python3 train.py
input
label
prediction