我想在Caffe上使用一些ConvNet运行推理。唯一的问题是我已经有了原始格式的权重和偏差,我不想在 Caffe 上重新训练它。
现在 Caffe 需要两个输入文件:
- .prototxt (用于网络信息(
- .caffemodel(用于权重和偏差(
在这里,.prototxt文件可以很容易地生成,但是可以使用原始权重和偏差生成.caffemodel吗?
手动操作Caffe模型的过程称为Net Surgery。
具体来说,net.params()
允许您直接访问网络的参数。