[Caffe]:關於編寫specific layer

實現 specific layer

1. 根據標準編寫specific_layer.hpp（存放在include/caffe/layers/目錄下）
2. 按照caffe標準模板編寫specific_layer.cpp（存放在src/caffe/layers/目錄下）
3. 實現cuda版本，即specific_layer.cu（同上）
4. 編寫test文件

修改caffe.proto

如果specific_layer需要傳參數進來，需要定義specific_layer_param.

// caffe.proto
// NOTE
// Update the next available ID when you add a new LayerParameter field.
//
// LayerParameter next available layer-specific ID: 147 (last added: recurent_param)
message LayerParameter {
...
...
+ optional SpecificParameter specific_param= 147//ID:147是下一個可用的編號
...
}

message SpecificParameter {
    /* define your specific parameters in this field */
}

這樣，自己寫的特殊網絡層就被加入到caffe中了，可以去編譯看看。

——————————————————————————————

後續將用實例介紹如何實現特殊層。