Questions about the BraTS2021 dataset

[yiyiyi36]

Hello! I'd like to ask whether the code is for binary classification on the BraTS2021 dataset? Is there code for multi-class segmentation? Thank you very much! @YipengHu @mathpluscode

[mathpluscode]

Hi @yiyiyi36 im terribly sorry that I only noticed this issue recently. The dataclass is defined in

ImgX-DiffSeg/imgx/datasets/brats2021_mr/brats2021_mr_dataset_builder.py

Line 129 in ce57729

BRATS2021_MR_INFO = BRATS2021MRDatasetInfo(

, it should be multi-class segmentation.