Learning Caffe(12)---deploy.prototxt文件

Deploy.prototxt文件簡述

       Deploy.prototxt是測試網絡是使用的文件,基於train.prototxt文件進行修改。

修改內容:

1.數據輸入

Deploy將train.prototxt中的數據層,包括訓練數據和測試數據刪除,新加入輸入層,用於輸入數據。

 

2.刪除參數初始化

Train.prototxt中會將weight,bias等參數進行初始化,在deploy中由於加載的是caffemodel中訓練好的參數,所以卷積層,全連接層的參數初始化要刪除。

3.刪除accuracy 層

在訓練過程中,用於輸出相對目標值的準確率,在deploy.prototxt中不需要該層。

 

4.修改softmaxwithLoss爲Softmax

     

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章