24 lines
508 B
Python
24 lines
508 B
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
import warnings # noqa: F401,F403
|
|
|
|
from mmcv.utils import Registry
|
|
|
|
TRANSFORMER = Registry('Transformer')
|
|
MASK_ASSIGNERS = Registry('mask_assigner')
|
|
MATCH_COST = Registry('match_cost')
|
|
|
|
|
|
def build_match_cost(cfg):
|
|
"""Build Match Cost."""
|
|
return MATCH_COST.build(cfg)
|
|
|
|
|
|
def build_assigner(cfg):
|
|
"""Build Assigner."""
|
|
return MASK_ASSIGNERS.build(cfg)
|
|
|
|
|
|
def build_transformer(cfg):
|
|
"""Build Transformer."""
|
|
return TRANSFORMER.build(cfg)
|