def build_block(cfg, **kwargs):
return build_object(cfg, BLOCKS, **kwargs)
def build_loss(cfg, **kwargs):
return build_object(cfg, LOSSES, **kwargs)
def build_embedder(cfg, **kwargs):
return build_object(cfg, EMBEDDERS, **kwargs)
def build_head(cfg, **kwargs):
return build_object(cfg, HEADS, **kwargs)
def build_model(cfg, **kwargs):
model = build_object(cfg, MODELS, **kwargs)
model = NNDataParallel(model, device_ids=[0])
return model
def build_detector(cfg, checkpoint, **kwargs):
model = build_object(cfg, DETECTORS, **kwargs)
load_checkpoint(model, checkpoint, map_location='cpu')
model = MMDataParallel(model, device_ids=[0])
model.eval()
return model
def build_dataset(cfg, **kwargs):
return build_object(cfg, DATASETS, **kwargs)