Caffe添加自定义层-自定义loss

Caffe最新版本，添加自定义层

1、实现自己的layer，另存相关的LossLayer的hpp和cpp文件为my_loss_layer.hpp，my_loss_layer.cpp，并修改对应的类名和构造函数，加入到对应的src(include)/caffe/layers，主要实现Setup、Forward_cpu、Backward_cpu。

2、如果需要GPU实现，那么在my_loss_layer.cu（同样在/src/caffe/layers）中实现Forward_gpu和Backward_gpu。

3、修改src/caffe/proto/caffe.proto，更新ID（proto文件里有个注释“// LayerParameter next available layer-specific ID: 145”，会告诉你可用的ID是145，添加自己的层后，修改为146，方便以后再次添加新的层）。在message LayerParameter域中，添加optional MyLossParameter my_loss_param = 145;最后添加

message MyLossParameter {

  // args1
  optional float args1 = 1 [default = 1.0];
  // args2
  optional float args2 = 2 [default = 0.01];

}

4、编译make all，只会编译修改的部分，所以编译会很快（如果之前已经编译好了）。

5、使用方法：

layer {
  name: "loss"
  type: "MyLoss"
  my_loss_param {//与在proto配置的名称要对应
    args1: 1
    args2: 2
  }
  bottom: "fc3"
  bottom: "label"
  top: "loss"
}