您的位置:首页 > Web前端

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

2018-07-30 16:19 337 查看

 

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

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐