Source code for fusionlab.functional.tfdice

import torch
import tensorflow as tf
from fusionlab.configs import EPS



def tf_dice_score(pred, target, axis=None):
    """
    Shape:
        - pred: :math:`(N, *, C)` where :math:`*` means any number of additional dimensions
        - target: :math:`(N, *, C)`, same shape as the input
        - Output: scalar.
    """
    intersection = tf.reduce_sum(pred * target, axis=axis)
    cardinality = tf.reduce_sum(pred + target, axis=axis)
    cardinality = tf.clip_by_value(cardinality,
                                   clip_value_min=EPS,
                                   clip_value_max=cardinality.dtype.max)
    return (2.0 * intersection) / cardinality