caffe - 通过pycaffe重命名Caffe层

Question

有没有一种简单的方法可以重命名 caffe 中的图层？使用 pycaffe 接口(interface)联网？

我浏览了net surgery示例，但我找不到我需要的示例。

例如，我想加载一个经过训练的 Caffe 模型并将名称更改为 conv1层及其对应的 blob 到 new-conv1 .

Answer 1

我不知道直接的方法，但这里有一个解决方法:

给定一个预训练的 Caffe 模型 my_model.caffemodel及其网络架构net.prototxt .复制 net.prototxt (比如 net_new.prototxt )，并更改 conv1 的名称层至new-conv1 (如果需要，您可以更改 bottom 和 top 的名称)。

import caffe
net_old = caffe.Net('net.prototxt','my_model.caffemodel',caffe.TEST)
net_new = caffe.Net('net_new.prototxt','my_model.caffemodel',caffe.TEST)
net_new.params['new-conv1'][0].data[...] = net_old.params['conv1'][0].data[...]  #copy filter across 2 nets
net_new.params['new-conv1'][1].data[...] = net_old.params['conv1'][1].data[...]  #copy bias
net_new.save('my_model_new.caffemodel')