__init__.py 382 B

12345678910111213141516
  1. import copy
  2. __all__ = ['build_metric']
  3. from .det_metric import DetMetric
  4. support_dict = ['DetMetric']
  5. def build_metric(config):
  6. config = copy.deepcopy(config)
  7. module_name = config.pop('name')
  8. assert module_name in support_dict, Exception(
  9. 'metric only support {}'.format(support_dict))
  10. module_class = eval(module_name)(**config)
  11. return module_class