Skip to content

GitLab

Unverified Commit 717d1571 authored by Yinhao Li's avatar Yinhao Li Committed by GitHub
Browse files

Add segmentation keys for greater_keys. (#1060)

parent e728608a
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 1 deletion
mmcv/runner/hooks/evaluation.py
View file @ 717d1571
...@@ -55,7 +55,10 @@ class EvalHook(Hook): ...@@ -55,7 +55,10 @@ class EvalHook(Hook):
rule_map = {'greater': lambda x, y: x > y, 'less': lambda x, y: x < y} rule_map = {'greater': lambda x, y: x > y, 'less': lambda x, y: x < y}
init_value_map = {'greater': -inf, 'less': inf} init_value_map = {'greater': -inf, 'less': inf}
greater_keys = ['acc', 'top', 'AR@', 'auc', 'precision', 'mAP'] greater_keys = [
'acc', 'top', 'AR@', 'auc', 'precision', 'mAP', 'mDice', 'mIoU',
'mAcc', 'aAcc'
]
less_keys = ['loss'] less_keys = ['loss']
def __init__(self, def __init__(self,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment