未验证 提交 f137929f 编写于 作者: W wangguanzhong 提交者: GitHub

refine focal loss (#23)

上级 c8a292b0
......@@ -389,6 +389,7 @@ class RetinaHead(object):
im_info=im_info,
num_classes=self.num_classes - 1)
fg_num = fluid.layers.reduce_sum(fg_num, name='fg_num')
score_tgt = fluid.layers.cast(score_tgt, 'int32')
loss_cls = fluid.layers.sigmoid_focal_loss(
x=score_pred,
label=score_tgt,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册