focal loss



        def focal_loss(prediction, target, weights=None, alpha=0.22, gamma=2):
            """Compute focal loss for predictions.
            Multi-labels Focal loss formula:
                FL = -alpha * (z-p)^gamma * log(p) -(1-alpha) * p^gamma * log(1-p)
                     ,which alpha = 0.25, gamma = 2, p = sigmoid(x), z = target_tensor.
            Args:
            prediction_tensor: A float tensor of shape [batch_size, num_anchors,
                num_classes] representing the predicted logits for each class
            target_tensor: A float tensor of shape [batch_size, num_anchors,
                num_classes] representing one-hot encoded classification targets
            weights: A float tensor of shape [batch_size, num_anchors]
            alpha: A scalar tensor for focal loss alpha hyper-parameter
            gamma: A scalar tensor for focal loss gamma hyper-parameter
            Returns:
            loss: A (scalar) tensor representing the value of the loss function
            """
            sigmoid_p = prediction
            print (alpha)
            zeros = array_ops.zeros_like(sigmoid_p, dtype=sigmoid_p.dtype) # Get a zero tensor with the given dtype.
            ones = tf.ones_like(sigmoid_p, dtype=sigmoid_p.dtype) #tf.constant(1.0, shape = prediction.get_shape())
            pos_p_sub = array_ops.where(target > zeros, ones - sigmoid_p, zeros)
            neg_p_sub = array_ops.where(target > zeros, zeros, sigmoid_p)


            per_entry_cross_ent = - alpha * (pos_p_sub ** gamma) * tf.log(tf.clip_by_value(sigmoid_p, 1e-8, 1.0)) \
                          - (1 - alpha) * (neg_p_sub ** gamma) * tf.log(tf.clip_by_value(1.0 - sigmoid_p, 1e-8, 1.0))
            return tf.reduce_sum(per_entry_cross_ent)

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值