从caffe 中提取部分层的权重,保存成.caffemodel,删除和修改caffe模型中任意最后一层或者任意层数网络的参数的方法 删除和修改caffe模型中任意最后一层或者任意层数网络的参数的方法

从caffe 中提取部分层的权重,保存成.caffemodel
根据deploy文件，需要的层加载需要的权重并保存

原

删除和修改caffe模型中任意最后一层或者任意层数网络的参数的方法

2016年12月05日 09:13:52 始于长虹 阅读数：3940

1、删除方法

在caffe中，很多训练完的模型只提取特征，然后比较两个特征的相似度，而不是分类。这个情况，可以删除caffe模型中的最后一层全连接层，这样可以大大减小模型，因为全连接层的参数非常多，方法如下：

 net = caffe.Net('XX_deploy.prototxt', 'XX.caffemodel', 'test');

 net.save('XX_remove_the_last_fc.caffemodel');

其实可以扩展到删除任意最后几层的参数，只需要在XX_deploy.prototxt中删除你需要删除层即可，呵呵，就这么简单。

2、修改方法

加载模型；net = caffe.Net('XX_deploy.prototxt', 'XX.caffemodel', 'test');

修改：net.layers('names').params(1).set_data(w);
           net.layers('names').params(2).set_data(b);

保存模型net.save('XX.caffemodel');