Shortcuts

Source code for mmpose.core.evaluation.eval_hooks

# Copyright (c) OpenMMLab. All rights reserved.
import warnings

from mmcv.runner import DistEvalHook as _DistEvalHook
from mmcv.runner import EvalHook as _EvalHook

MMPOSE_GREATER_KEYS = [
    'acc', 'ap', 'ar', 'pck', 'auc', '3dpck', 'p-3dpck', '3dauc', 'p-3dauc',
    'pcp'
]
MMPOSE_LESS_KEYS = ['loss', 'epe', 'nme', 'mpjpe', 'p-mpjpe', 'n-mpjpe']


[docs]class EvalHook(_EvalHook): def __init__(self, dataloader, start=None, interval=1, by_epoch=True, save_best=None, rule=None, test_fn=None, greater_keys=MMPOSE_GREATER_KEYS, less_keys=MMPOSE_LESS_KEYS, **eval_kwargs): if test_fn is None: from mmpose.apis import single_gpu_test test_fn = single_gpu_test # to be compatible with the config before v0.16.0 # remove "gpu_collect" from eval_kwargs if 'gpu_collect' in eval_kwargs: warnings.warn( '"gpu_collect" will be deprecated in EvalHook.' 'Please remove it from the config.', DeprecationWarning) _ = eval_kwargs.pop('gpu_collect') # update "save_best" according to "key_indicator" and remove the # latter from eval_kwargs if 'key_indicator' in eval_kwargs or isinstance(save_best, bool): warnings.warn( '"key_indicator" will be deprecated in EvalHook.' 'Please use "save_best" to specify the metric key,' 'e.g., save_best="AP".', DeprecationWarning) key_indicator = eval_kwargs.pop('key_indicator', 'AP') if save_best is True and key_indicator is None: raise ValueError('key_indicator should not be None, when ' 'save_best is set to True.') save_best = key_indicator super().__init__(dataloader, start, interval, by_epoch, save_best, rule, test_fn, greater_keys, less_keys, **eval_kwargs)
[docs]class DistEvalHook(_DistEvalHook): def __init__(self, dataloader, start=None, interval=1, by_epoch=True, save_best=None, rule=None, test_fn=None, greater_keys=MMPOSE_GREATER_KEYS, less_keys=MMPOSE_LESS_KEYS, broadcast_bn_buffer=True, tmpdir=None, gpu_collect=False, **eval_kwargs): if test_fn is None: from mmpose.apis import multi_gpu_test test_fn = multi_gpu_test # to be compatible with the config before v0.16.0 # update "save_best" according to "key_indicator" and remove the # latter from eval_kwargs if 'key_indicator' in eval_kwargs or isinstance(save_best, bool): warnings.warn( '"key_indicator" will be deprecated in EvalHook.' 'Please use "save_best" to specify the metric key,' 'e.g., save_best="AP".', DeprecationWarning) key_indicator = eval_kwargs.pop('key_indicator', 'AP') if save_best is True and key_indicator is None: raise ValueError('key_indicator should not be None, when ' 'save_best is set to True.') save_best = key_indicator super().__init__(dataloader, start, interval, by_epoch, save_best, rule, test_fn, greater_keys, less_keys, broadcast_bn_buffer, tmpdir, gpu_collect, **eval_kwargs)
Read the Docs v: latest
Versions
latest
1.x
v0.14.0
fix-doc
cn_doc
Downloads
pdf
html
epub
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.