【发布时间】:2017-12-11 05:44:52
【问题描述】:
我想在 Caffe 上使用一些 ConvNet 进行推理。唯一的问题是我已经有了原始格式的权重和偏差,我不想在 Caffe 上重新训练它。
现在 Caffe 需要两个输入文件:
- .prototxt(用于网络信息)
- .caffemodel(用于权重和偏差)
这里可以轻松生成.prototxt文件,但是是否可以使用原始权重和偏差生成.caffemodel?
【问题讨论】:
我想在 Caffe 上使用一些 ConvNet 进行推理。唯一的问题是我已经有了原始格式的权重和偏差,我不想在 Caffe 上重新训练它。
现在 Caffe 需要两个输入文件:
这里可以轻松生成.prototxt文件,但是是否可以使用原始权重和偏差生成.caffemodel?
【问题讨论】:
手动操作 Caffe 模型的过程称为Net Surgery。
具体来说,net.params() 允许您直接访问网络的参数。
【讨论】: