caffe下添加新的loss层

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,编译通过即可。

如果喜欢请点个赞再走哦,有什么问题请留言区讨论,一起学习。

评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值