caffemodel的读取与修改

直接撸代码~

 import caffe
 import numpy as np

 caffe.set_mode_cpu()
 net = caffe.Net('myprototxt.prototxt', 'mycaffemodel.caffemodel', caffe.TEST)
 netNew = caffe.Net('myprototxt2.prototxt', 'mycaffemodel2.caffemodel', caffe.TEST)
 for k, v in net.params.items():
     #print (k, v[0].data.shape)
     #print np.size(net.params[k])
     for i in range(np.size(net.params[k])):
         netNew.params[k][i].data[:] = np.copy(net.params[k][i].data[:])
 params = {('layer_name_new','layer_name_old')}
 for k, v in params:
     for i in range(np.size(net.params[v])):
         netNew.params[k][i].data[:] = np.copy(net.params[v][i].data[:])
 netNew.save('mynewmodel.caffemodel')