2017-07-07 42 views
0

我想使用Caffe上的一些ConvNet來運行推理。唯一的問題是我已經有了原始格式的權重和偏見,我不想在Caffe上重新訓練它。如何在Caffe中運行非Caffe模型

現在來自Caffe需要兩個輸入文件:

  • .prototxt(用於網絡信息)
  • .caffemodel(權重爲和偏置)

這裏可以輕鬆生成.prototxt文件,但是有可能使用原始權重和偏差來生成.caffemodel嗎?

回答

1

手動操作Caffe模型的過程被稱爲Net Surgery

具體而言,net.params()允許您直接訪問網絡的參數。