f29fe9adc0cfd99ae0689f548d13f9822baae39f,kornia/losses/focal.py,,focal_loss,#Any#Any#Any#Any#Any#Any#,13

Before Change

                        .format(type(input)))

    if not len(input.shape) == 4:
        raise ValueError("Invalid input shape, we expect BxNxHxW. Got: {}"
                         .format(input.shape))

    if not input.shape[-2:] == target.shape[-2:]:
        raise ValueError("input and target shapes must be the same. Got: {}"

After Change

                         .format(input.shape))

    if input.size(0) != target.size(0):
        raise ValueError("Expected input batch_size ({}) to match target batch_size ({})."
                         .format(input.size(0), target.size(0)))

    n = input.size(0)
    out_size = (n,) + input.size()[2:]

Instances