caffe的使用当中,最神秘的就是caffe的源码,而我们要做一些创新点的话,基本上都是要修改caffe的源码。
小鱼和大家分享怎么在caffe下添加新的层。
比如要添加一个新的loss函数:euclidean_loss_norm_layer
1、需要编写相应的euclidean_loss_norm_layer.cpp和相应的euclidean_loss_norm_layer.cu文件,.cpp文件是实现cpu下函数的前传与反传,.cu文件是实现gpu下函数的前传与反传;
2、将euclidean_loss_norm_layer.cpp和
euclidean_loss_norm_layer.cu文件放入caffe/src/caffe/layers文件夹下;
3、编写对应的euclidean_loss_norm_layer.hpp文件,放入caffe/include/caffe/layers文件夹下;
4、在caffe/src/caffe/proto/caffe.proto文件下添加对应的optional及meaasge,如下:
optional EuclideanLossNormParameter euclidean_loss_norm_param = 149;
message EuclideanLossNormParameter {
// The axis along which to concatenate -- may be negative to index from the
// end (e.g., -1 for the last axis). Other axes must have the
// same dimension for all the bottom blobs.
// By default, ConcatLayer concatenates blobs along the "channels" axis (1).
// optional int32 axis = 2 [default = 1];
// DEPRECATED: alias for "axis" -- does not support negative indexing.
//optional uint32 concat_dim = 1 [default = 1];
}
5、重新在caffe下make clean及make all,编译通过即可。
如果喜欢请点个赞再走哦,有什么问题请留言区讨论,一起学习。