search

zimenglan-sysu-512 / Focal-Loss

loss layer of implementation
125 stars 41 forks source link

when have two classes, it not work. #16

Open ngunauj opened 5 years ago

ngunauj commented 5 years ago

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.

1343464520 commented 4 years ago

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