提取MatConvnet模型權重

Question

我目前正在開發面部識別應用程序。

該算法是使用MatConvnet庫（http://www.vlfeat.org/matconvnet/）實現和訓練的。最後，我有一個網絡（.MAT文件），它看起來像這樣：

我想知道如果有可能利用其.MAT文件來提取網絡的權重，將它們寫入XML文件並用Caffe C++讀取它們。我想在Caffe C++中重用它們來做一些測試和硬件實現。有沒有一種有效和實用的方法來進行？

非常感謝您的幫助。

Answer 1

有一個轉換腳本可以將matconvnet模型轉換爲咖啡模型here，您可能會發現它很有用。

Answer 2

您想要存儲參數的圖層必須設置爲「珍貴」。在net.var中，您可以訪問參數並寫入它們。

Answer 3

您不能使用matconvnet的訓練網絡的權重咖啡。您只需將您的模型從matconvnet導入caffe。（https://github.com/vlfeat/matconvnet/blob/4ce2871ec55f0d7deed1683eb5bd77a8a19a50cd/utils/import-caffe.py）。但是這個腳本不支持所有的圖層，你可能在使用它時遇到困難。 最好的方法是在python中定義caffe prototxt作爲matconvnet模型。