Python gen_init_net_from_blobs Examples

Example #1

File: convert_pkl_to_pb.py Project: darthsuogles/Detectron

def gen_init_net(net, blobs, empty_blobs):
    blobs = copy.deepcopy(blobs)
    for x in empty_blobs:
        blobs[x] = np.array([], dtype=np.float32)
    init_net = mutils.gen_init_net_from_blobs(blobs, net.external_inputs)
    init_net = core.Net(init_net)
    return init_net

Example #2

File: convert_pkl_to_pb.py Project: Jhird/Detectron

def gen_init_net(net, blobs, empty_blobs):
    blobs = copy.deepcopy(blobs)
    for x in empty_blobs:
        blobs[x] = np.array([], dtype=np.float32)
    init_net = mutils.gen_init_net_from_blobs(
        blobs, net.external_inputs)
    init_net = core.Net(init_net)
    return init_net