mirror of
https://github.com/alibaba/EasyCV.git
synced 2025-06-03 14:49:00 +08:00
21 lines
601 B
Python
21 lines
601 B
Python
from easycv.utils.registry import Registry, build_from_cfg
|
|
|
|
BBOX_ASSIGNERS = Registry('bbox_assigner')
|
|
BBOX_SAMPLERS = Registry('bbox_sampler')
|
|
BBOX_CODERS = Registry('bbox_coder')
|
|
|
|
|
|
def build_bbox_assigner(cfg, **default_args):
|
|
"""Builder of box assigner."""
|
|
return build_from_cfg(cfg, BBOX_ASSIGNERS, default_args)
|
|
|
|
|
|
def build_bbox_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)
|