focalloss的caffe复现版 -ag凯发k8国际

原代码见： 
https://github.com/chuanqi305/focalloss

1，caffe.proto 
源文件在src/caffe/proto/目录里 
从492行这些optional里，作者添加了两行:

optional relu6parameter relu6_param = 208;
optional focallossparameter focal_loss_param = 147;

从895行这里添加了一行:

optional bool half_pad = 19 [default = false];

从1425行这里添加一行:

optional bool reduce_boxes = 14 [default = false];

从1505行添加了一段:

message relu6parameter{
    enum engine {
        default = 0;
        cafee = 1;
        cudnn = 2;
    }
    optional engine engine = 2[default = default];
}

从1641行添加一段:

message focallossparameter{
    enum engine{
        default = 0;
        caffe = 1;
        cudnn = 2;
    }
    optional engine engine = 1[default = default];

    //the axis along which to perform the softmax -- may be negative to index
    //from the end(e.g., -1 for the last axis).
    //any other axes will be evaluated as independent softmaxes.
    optional int32 axis = 2[default = 1];
    optional float alpha = 3[default = 0.25];
    optional float gamma = 4[default = 2.0];
}

2.在src/caffe/layers/下放入focal_loss_layer.cpp和focal_loss_layer.cu文件

3.在include/caffe/layers/下放入focla_loss_layer.hpp

重新编译caffe。
--------------------- 
作者：苏格兰橘猫 
来源：csdn 
原文：https://blog.csdn.net/qq_34951080/article/details/78491009 
ag凯发k8国际的版权声明：本文为博主原创文章，转载请附上博文链接！

与50位技术专家面对面20年技术见证，附赠技术全景图

总结

以上是ag凯发k8国际为你收集整理的focalloss的caffe复现版的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得ag凯发k8国际网站内容还不错，欢迎将ag凯发k8国际推荐给好友。