-
大小: 4KB文件類型: .zip金幣: 1下載: 0 次發布日期: 2021-01-04
- 語言: 其他
- 標簽:
資源簡介
論文中"Holistically-Nested Edge Detection"使用的加權損失函數,具體用法見博客http://blog.csdn.net/majinlei121/article/details/78884531
代碼片段和文件信息
#include?
#include?
#include?“caffe/layers/sigmoid_cross_entropy_loss_layer.hpp“
#include?“caffe/util/math_functions.hpp“
namespace?caffe?{
template?
void?SigmoidCrossEntropyLosslayer::layerSetUp(
????const?vector*>&?bottom?const?vector*>&?top)?{
??Losslayer::layerSetUp(bottom?top);
??sigmoid_bottom_vec_.clear();
??sigmoid_bottom_vec_.push_back(bottom[0]);
??sigmoid_top_vec_.clear();
??sigmoid_top_vec_.push_back(sigmoid_output_.get());
??sigmoid_layer_->SetUp(sigmoid_bottom_vec_?sigmoid_top_vec_);
??has_ignore_label_?=
????this->layer_param_.loss_param().has_ignore_label();
??if?(has_ignore_label_)?{
????ignore_label_?=?this->layer_param_.loss_param().ignore_label();
??}
??if?(this->layer_param_.loss_param().ha
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2017-12-24?15:11??SigmoidCrossEntropyLoss\
?????文件????????5443??2017-12-24?04:42??SigmoidCrossEntropyLoss\sigmoid_cross_entropy_loss_la
?????文件????????7388??2017-11-20?08:06??SigmoidCrossEntropyLoss\sigmoid_cross_entropy_loss_la
評論
共有 條評論