analysis.py 文件源码

def _f_dice(a, b):
    """DICE between two segmentations.

    Args:
        a: [..., H, W], binary mask
        b: [..., H, W], binary mask

    Returns:
        dice: [...]
    """
    card_a = a.sum(axis=-1).sum(axis=-1)
    card_b = b.sum(axis=-1).sum(axis=-1)
    card_ab = (a * b).sum(axis=-1).sum(axis=-1)
    card_sum = card_a + card_b
    dice = 2 * card_ab / (card_sum + np.equal(card_sum, 0).astype('float32'))
    return dice