首先要在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调用这个类
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:caffe实现focal loss层的一些理解和对实现一个layer层易犯错的地方的总结 - Python技术站