zimenglan-sysu-512/Focal-Loss

when have two classes, it not work.

Opened this issue · 1 comments

layer {
name: "loss_cls_pos"
type: "FocalLoss"
bottom: "rpn_cls_score_reshape"
bottom: "labels"
propagate_down: 1
propagate_down: 0
top: "loss_cls_pos"
include { phase: TRAIN }
loss_weight: 1
loss_param { ignore_label: -1 normalize: true }
focal_loss_param { alpha: 0.75 gamma: 2 }
}

i have 2 classes -1 and 1. the loss is ok. but it can't classify.

@ngunauj do you solve it? i also have the same problem...