a dp@sdddlZddlmZddlmmZddlmZd ejeje ejdddZ Gdddej Z dS) N)one_hot:0yE>)inputtargetepsreturnc Cst|tjstdt|t|jdks>> N = 5 # num_classes >>> input = torch.randn(1, N, 3, 5, requires_grad=True) >>> target = torch.empty(1, 3, 5, dtype=torch.long).random_(N) >>> output = dice_loss(input, target) >>> output.backward() z&Input type is not a torch.Tensor. Got z-Invalid input shape, we expect BxNxHxW. Got: Nz/input and target shapes must be the same. Got: z and z2input and target must be in the same device. Got: )Zdim)Z num_classesdevicedtype)r g@g?) isinstancetorchTensor TypeErrortypelenshape ValueErrorr FZsoftmaxrr summean) rrrZ input_softZtarget_one_hotdims intersectionZ cardinalityZ dice_scorer[/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/losses/dice.py dice_loss s&  rcsBeZdZdZd eddfdd Zejejejddd ZZ S) DiceLossuWCriterion that computes Sørensen-Dice Coefficient loss. According to [1], we compute the Sørensen-Dice Coefficient as follows: .. math:: \text{Dice}(x, class) = \frac{2 |X| \cap |Y|}{|X| + |Y|} Where: - :math:`X` expects to be the scores of each class. - :math:`Y` expects to be the one-hot tensor with the class labels. the loss, is finally computed as: .. math:: \text{loss}(x, class) = 1 - \text{Dice}(x, class) Reference: [1] https://en.wikipedia.org/wiki/S%C3%B8rensen%E2%80%93Dice_coefficient Args: eps: Scalar to enforce numerical stabiliy. Shape: - Input: :math:`(N, C, H, W)` where C = number of classes. - Target: :math:`(N, H, W)` where each value is :math:`0 ≤ targets[i] ≤ C−1`. Example: >>> N = 5 # num_classes >>> criterion = DiceLoss() >>> input = torch.randn(1, N, 3, 5, requires_grad=True) >>> target = torch.empty(1, 3, 5, dtype=torch.long).random_(N) >>> output = criterion(input, target) >>> output.backward() rN)rrcst||_dSN)super__init__r)selfr __class__rrr"ts zDiceLoss.__init__)rrrcCst|||jSr )rr)r#rrrrrforwardxszDiceLoss.forward)r) __name__ __module__ __qualname____doc__floatr"rrr& __classcell__rrr$rrMs&r)r) rZtorch.nnnnZtorch.nn.functionalZ functionalrZkornia.utils.one_hotrrr+rModulerrrrrs   B