项目作者: owphoo

项目描述 :
focal loss caffe version
高级语言: C++
项目地址: git://github.com/owphoo/FocalLoss.git
创建时间: 2018-08-05T08:38:28Z
项目社区:https://github.com/owphoo/FocalLoss

开源协议:

下载


FocalLoss

focal loss caffe version

Usage

  1. Update your caffe.proto and the code.
  2. make -j4
  3. usage of FocalLoss layer
    ```

    if your task is a 3-class classification problem

    layer{
    name: “loss”
    type: “FocalLoss”
    bottom: “feature”
    bottom: “label”
    top: “loss”
    include {
    1. phase: TRAIN
    }
    focal_loss_param {
    1. alpha_balance: true
    2. gamma: 2
    3. alpha: 0.5 #background
    4. alpha: 1
    5. alpha: 1
    }
    }

```

Derivation

equation

Reference

Tsung-Yi Lin, Priya Goyal, Ross Girshick, Kaiming He, Piotr Dollár, Focal Loss for Dense Object Detection, Paper Link