mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add segmentation keys for greater_keys. (#1060)

pull/1067/head
Yinhao Li 2021-05-31 10:13:55 +08:00 committed by GitHub
parent e728608ac9
commit 717d15719c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mmcv/runner/hooks/evaluation.py
View File

@ -55,7 +55,10 @@ class EvalHook(Hook):
rule_map = {'greater': lambda x, y: x > y, 'less': lambda x, y: x < y}
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']
def __init__(self,