-
大小: 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
评论
共有 条评论