首先要在caffe.proto中的LayerParameter中增加一行optional FocalLossParameter focal_loss_param = 205;,然后再单独在caffe.proto中增加
message FocalLossParameter{ optional float gamma = 1 [default = 2]; optional float alpha = 2 [default = 0.25];
}
实际上可以这样理解,FocalLossParameter
就是一个layer下的子类,focal_loss_param
相当于这个类的一个实例,message FocalLossParameter
就相当于这个类的类定义
只要在hpp中引用#include "caffe/proto/caffe.pb.h",就可以在cpp调用这个类