* add dice evaluation metric * add dice evaluation metric * add dice evaluation metric * support 2 metrics * support 2 metrics * support 2 metrics * support 2 metrics * fix docstring * use np.round once for all
9 lines
273 B
Python
9 lines
273 B
Python
from .class_names import get_classes, get_palette
|
|
from .eval_hooks import DistEvalHook, EvalHook
|
|
from .metrics import eval_metrics, mean_dice, mean_iou
|
|
|
|
__all__ = [
|
|
'EvalHook', 'DistEvalHook', 'mean_dice', 'mean_iou', 'eval_metrics',
|
|
'get_classes', 'get_palette'
|
|
]
|