Caffe Loss层 - HingelossLayer

AIHGF

修改于 2020-06-12 11:11:27

4500

修改于 2020-06-12 11:11:27

文章被收录于专栏：AIUAI AIUAIAIUAI

HingelossLayer

计算 one-of-many 分类任务的 hinge loss.

Hinge Loss 概念

Caffe prototxt定义

......
layer {
  name: "fc8voc"
  type: "InnerProduct"
  bottom: "fc7"
  top: "fc8voc"
  param {
    lr_mult: 10
    decay_mult: 1
  }
  param {
    lr_mult: 2
    decay_mult: 0
  }
  inner_product_param {
    num_output: 20
    weight_filler {
      type: "gaussian"
      std: 0.01
    }
    bias_filler {
      type: "constant"
      value: 0
    }
  }
}

layer {
  name: "loss"
  type: "HingeLossMultiLabel"
  bottom: "fc8voc"
  bottom: "label"
  top: "loss"
}