dac6839364405ff932bc33a5ca7845cba4f8b4ce,kornia/color/adjust.py,,adjust_brightness,#Any#Any#,31

Before Change

            type(image)))

    if len(image.shape) < 3:
        raise ValueError("Input size must have a shape of (*, H, W). Got {}"
                         .format(image.shape))

    if (brightness_factor < torch.zeros(1)).any():
        raise ValueError("Brightness factor must be non-negative. Got {}"
                         .format(brightness_factor))

After Change

        raise TypeError(f"The factor should be either a float or torch.Tensor. "
                        f"Got {type(brightness_factor)}")

    if isinstance(brightness_factor, float):
        brightness_factor = torch.tensor([brightness_factor])

    brightness_factor = brightness_factor.to(input.device).to(input.dtype)

    if (brightness_factor < torch.zeros(1)).any():
        raise ValueError(f"Brightness factor must be non-negative. Got {brightness_factor}")

Instances