MRM-pytorch
MRM-pytorch copied to clipboard
KeyError: 'Dice. front' when training the segementation task
Thank you for sharing the code!
There is an error occur when I trainning the segementation task: File ".../site-packages/mmcv/runner/hooks/evaluation.py", line 389, in evaluate return eval_res[self.key_indicator] KeyError: 'Dice.front'
After inspection, I found that in /mmseg/core/evaluation/metrics. py, the function 'total_area_to_metrics' has the following definition:
Allowed_ Metrics=['mIoU ',' mDice ',' mFscore ',' medDice '] But there is no 'if' item for 'medDice': For metric in metrics: If metric=='mIoU ': ... If metric=='mDice': ... I guess is any code missing here? or can I directly use [If metric=='mDice' or metric=='medDice ': ] ?