File size: 570 Bytes
b0481db |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
from utils.config import Config, ConfigDict
from rscd.models.backbones import *
from rscd.models.decoderheads import *
from rscd.losses import *
def build_from_cfg(cfg):
if not isinstance(cfg, (dict, ConfigDict, Config)):
raise TypeError(
f'cfg should be a dict, ConfigDict or Config, but got {type(cfg)}')
if 'type' not in cfg:
raise KeyError(
'`cfg` must contain the key "type", '
f'but got {cfg}')
obj_type = cfg.pop('type')
obj_cls = eval(obj_type)
obj = obj_cls(**cfg)
return obj |