这是《Multi-task learning with residual attention for brain tumor segmentation and classification》这篇文章的源代码。
请先将data文件夹按照您需要的比例对数据进行划分,之后输入$python main.py train$进行模型训练,输入$python main.py test$进行模型测试。
数据集来源:https://figshare.com/articles/dataset/brain_tumor_dataset/1512427
This is the source code for the article "Multi-task learning with residual attention for brain tumor segmentation and classification".
First, divide the data folder into training set, validation set, and test set in the scale you need. After that, enter "python main.py train" for model training and "python main.py test" for model testing.
Data set source: https://figshare.com/articles/dataset/brain_tumor_dataset/1512427