Esempi in Python per load_model_pb

Esempio n. 1

File: get_runtime.py Progetto: zhangkehua/mobile-vision

def load_pb(model_path):
    # Load a caffe2 pb model and generate blobs
    model_path = os.path.join(model_path, "model.pb")
    model_init_path = (model_path + "/model_init.pb") \
        if os.path.exists(model_path + "/model_init.pb") else None

    model, _ = model_utils.load_model_pb(model_path,
                                         model_init_path,
                                         is_run_init=True)

    return model

Esempio n. 2

File: eval_seg_cpu.py Progetto: dreiss/maskrcnn-nnapi

def load_model(args):
    net, _ = model_utils.load_model_pb(args.net, args.init_net)
    return net

Esempio n. 3

def load_model(args):
    net, _ = model_utils.load_model_pb(args.net,
                                       args.init_net,
                                       is_run_init=True,
                                       is_create_net=True)
    return net