u-net-brain-tumor
u-net-brain-tumor copied to clipboard
View brain image and segmented image.
How we can view the brain MRI and finally the segmented image.
https://github.com/Lafite-Yu/BraTS_2018_U-Net/blob/master/run.py
Thank you Lafite-Yu for the code. I was able to successfully execute it. I have generated the .npz model for enhance and necrotic tumor. How can I know that which type of Tumor it is by executing the code on an input file of type .nii.gz
According to the readme document of this project, you should train one model for one task, relatively. And I didn't change this, you can change the code in train.py
or use command like python train.py --task edema
to train to separate the edema part of the tumor.
parser.add_argument('--task', type=str, default='all', help='all, necrotic, edema, enhance')
# there are 4 labels in targets:
# Label 0: background
# Label 1: necrotic and non-enhancing tumor
# Label 2: edema
# Label 4: enhancing tumor
You can see know more about that in train.py
.
There are four values in the segmentation results of the BraTS dataset, shown as previous comment.