OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
21 lines
628 B
21 lines
628 B
# Copyright (c) OpenMMLab. All rights reserved. |
|
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)
|
|
|