コード例 #1
0
def load_model(args):
    model = test_engine.initialize_model_from_cfg(args.wts_file)
    blobs = mutils.get_ws_blobs()

    return model, blobs
コード例 #2
0
def load_model(args):
    model = test_engine.initialize_model_from_cfg(cfg.TEST.WEIGHTS)
    blobs = mutils.get_ws_blobs()

    return model, blobs
コード例 #3
0
ファイル: convert_pkl_to_pb.py プロジェクト: Jhird/Detectron 
def load_model(args):
    model = test_engine.initialize_model_from_cfg(cfg.TEST.WEIGHTS)
    blobs = mutils.get_ws_blobs()

    return model, blobs