caffe測試--如何得到deploy.prototxt?

 

步驟:

 

  1. 複製train_val.prototxt中的所有層。
  2. 刪除沒有“bottom”字段的前幾層(用於數據輸入的data層 train test),並設置新的data層(不包含train、test,不指定類型,指定數據輸入的格式input_dim)。圖1
  3. 刪除所有必須專門學習的參數。blobs_lr,weight_decay,weight_filler,bias_filler 圖2
  4. 修改softmax層的類型,在訓練網絡時使用的是SoftmaxWithLoss返回loss,現在需要獲得預測的結果,所以使用prob  圖3

圖1

圖2

圖3

參考:https://github.com/BVLC/caffe/issues/550

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