caffe实现focal loss层的一些理解和对实现一个layer层易犯错的地方的总结

 

首先要在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调用这个类