mirror of
https://github.com/deepinsight/insightface.git
synced 2026-05-18 14:55:42 +00:00
21 lines
580 B
Python
Executable File
21 lines
580 B
Python
Executable File
from mmcv.utils import Registry, build_from_cfg
|
|
|
|
BBOX_ASSIGNERS = Registry('bbox_assigner')
|
|
BBOX_SAMPLERS = Registry('bbox_sampler')
|
|
BBOX_CODERS = Registry('bbox_coder')
|
|
|
|
|
|
def build_assigner(cfg, **default_args):
|
|
"""Builder of box assigner."""
|
|
return build_from_cfg(cfg, BBOX_ASSIGNERS, default_args)
|
|
|
|
|
|
def build_sampler(cfg, **default_args):
|
|
"""Builder of box sampler."""
|
|
return build_from_cfg(cfg, BBOX_SAMPLERS, default_args)
|
|
|
|
|
|
def build_bbox_coder(cfg, **default_args):
|
|
"""Builder of box coder."""
|
|
return build_from_cfg(cfg, BBOX_CODERS, default_args)
|