原代碼見：?
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?
版權(quán)聲明：本文為博主原創(chuàng)文章，轉(zhuǎn)載請附上博文鏈接！

與50位技術(shù)專家面對面20年技術(shù)見證，附贈技術(shù)全景圖

總結(jié)

以上是生活随笔為你收集整理的FocalLoss的Caffe复现版的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。