在新版本caffe裡新增新的一層(向舊格式中加)

2021-08-21 00:08:09 字數 729 閱讀 7755

今天實現了在caffe框架中加入乙個層,完成歐式距離的任務。之所以這樣,是因為還沒有實現自己想要的loss,只是試著學者,看能不能把新增層的流程順下來。最後實現了。

(一)總體框架

1)在./src/caffe/proto/caffe.proto 中增加 對應layer的paramter message;查詢euclidean,然後根據euclidean_loss加乙個whole_loss,加乙個數就可以了。

2)在./include/caffe/layers/ 資料夾中增加該layer的類的宣告.h檔案,直接將歐式距離的hpp檔案euclidean_loss_layer.h複製,命名為whole_loss_layer.h,然後開啟檔案,將euclidean替換為whole,注意大小寫。

3)在./src/caffe/layers/目錄下新建.cpp和.cu檔案,進行類實現。也是複製,然後改檔名,替換檔案中的euclidean,注意大小寫。

4)在./src/caffe/gtest/中增加layer的測試**,對所寫的layer前傳和反傳進行測試,測試還包括速度。

最後一步很多人省了,或者沒意識到,但是為保證**正確,建議還是嚴格進行測試,磨刀不誤砍柴功。

(二)編譯

make all -j 8

(三)測試

找乙個工程**,將**中的network定義檔案中的euclidean_loss_layer層換為whole_loss_layer層

注意:一定要注意細節,大小寫

caffe新版本的各種軟體

系統重灌了,於是,我想裝就體驗一下最新的各種東西吧。anaconda最新的 cuda最新的 cudnn最新的 本來安裝好了沒問題。caffe編譯也通過了。但是不能用,缺少python opencv和python protobuf,於是用conda裝,這個時候就出現各種問題了。開始是opencv不能用...

Wamp整合環境 新增PHP的新版本

首先我的wamp整合環境是5.4.12環境 然後二次開發乙個5.5版本以上的 寫法的問題,如圖 執行錯誤。無奈要安裝php5.5版本以後的。密碼 40la 我的wamp apache為2.4版本 php為5.4.12版本 在原來版本裡面找到wampserver.conf 這個檔案複製乙份到新版本的p...

Unity引擎新版本 讓開發者在VR裡開發VR遊戲

unity引擎新版本 讓開發者在vr裡開發vr遊戲。各大遊戲引擎紛紛開始支援開發 vr 遊戲已經不是什麼新鮮事了,作為刪手機遊戲界扛把子的引擎 unity 自然也不能落後。但這一次他們的思路有些清奇 讓遊戲開發者在 vr 裡開發 vr 遊戲。12 月 16 日 unity 發布的 editorvr ...