def _get_predictor()

in source/inference.py [0:0]

• 10 lines of code
• 2 McCabe index (conditional complexity)

def _get_predictor(config_path, model_path):
    
    cfg = get_cfg()
    
    cfg.merge_from_file(config_path) # get baseline parameters from YAML config
    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.20  # set threshold for this model
    cfg.MODEL.WEIGHTS = model_path
    cfg.DATASETS.TEST = ("cb_val", )

    pred = DefaultPredictor(cfg)
    logger.info(cfg)
    eval_results = pred.model.eval()

    return pred