| 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) | |