Python build_object Exemples

Exemple #1

def build_block(cfg, **kwargs):
    return build_object(cfg, BLOCKS, **kwargs)

Exemple #2

def build_loss(cfg, **kwargs):
    return build_object(cfg, LOSSES, **kwargs)

Exemple #3

def build_embedder(cfg, **kwargs):
    return build_object(cfg, EMBEDDERS, **kwargs)

Exemple #4

def build_head(cfg, **kwargs):
    return build_object(cfg, HEADS, **kwargs)

Exemple #5

def build_model(cfg, **kwargs):
    model = build_object(cfg, MODELS, **kwargs)
    model = NNDataParallel(model, device_ids=[0])
    return model

Exemple #6

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

Exemple #7

def build_dataset(cfg, **kwargs):
    return build_object(cfg, DATASETS, **kwargs)