search

himashi92 / vizviva_brats_2021

[Brainlesion 2021] Official PyTorch Implementation for Reciprocal Adversarial Learning for Brain Tumor Segmentation: A Solution to BraTS Challenge 2021 Segmentation Task
MIT License
20 stars 3 forks source link

Test Phase Dice Scores #12

Closed tarekegn82 closed 10 months ago

tarekegn82 commented 11 months ago

Dear @himashi92

After running test experiments, I only found segmentation mask under segs folder. I would like to know how to get the dice scores of each tumor class as presented in this paper. Please kindly respond with the necessary details.

thank you

himashi92 commented 11 months ago

Hi,

The test.py script is written in a way that handles the inference of the online brats challenge where we do not get to see ground truths. If you have groundtruths for the test dataset, use the generate_segmentations function in the utils.py script: *https://github.com/himashi92/vizviva_brats_2021/blob/main/utils.py#L218**.

Cheers, Himashi

tarekegn82 commented 11 months ago

writer After importing generate_segmentations from the utils.py file, I tried to use the function inside test.py main function. The writer is not defined error occurs when I include it inside the generate_segmentations function inside the test.py main. Also, when I remove it, the attached error occurs. Are there any other suggestions that can help me resolve this error? also can you provide an updated test.py when you have time, please? @himashi92