net_surgery.py 文件源码

def pre_transplant(inp_proto, inp_model, fcn_proto, fcn_model):
    state = caffe.TRAIN
    net = caffe.Net(inp_proto, inp_model, state)

    new_net = caffe.Net(fcn_proto, inp_model, state)

    new_net = transplant(new_net, net, '-deconv')
    new_net.save(fcn_model)