• 大小: 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_layer.hpp
     文件        7388  2017-11-20 08:06  SigmoidCrossEntropyLoss\sigmoid_cross_entropy_loss_layer.cpp

评论

共有 条评论

相关资源