caffe-windows添加層

標籤：檔案中   linux環境   自己   pool   ssi   https   .com   inux   複製   

最近接觸到Fine-Grained Classification, 其中的經典方法Bilinear CNN用到了Bilinear Layer, L2-normlize Layer, signed-sqrt Layer等

這些層在caffe-windows已經BVLC版caffe中是不存在的，若想應用的話需要將上述層添加到caffe工程

https://github.com/gy20073/compact_bilinear_pooling/tree/master/caffe-20160312 好在已經有大神們做了這些層的相關implementation.. 感謝

 

對於Linux環境下caffe的話之前做過相關層的添加，過程比較簡單安裝上述連結中的步驟走完即可，大體步驟：

1、src 中的cpp及cu檔案複製到自己caffe工程相對的src中

2、include中的hpp檔案複製到自己caffe相應的include中

3、src中 caffe.proto 檔案中添加新層的參數，注意序號不能與已存在層重複

因為BCNN還涉及到了cuda中的相關計算，所以還做了cuda的相關修改，見連結

 

對於caffe-windows則需要將步驟1,2中的檔案在vs的sln檔案中添加到相應的位置..否則產生的工程中還是沒有這些層的應用

caffe-windows添加層