def apply_sharpness(input: torch.Tensor, params: Dict[str, torch.Tensor]) -> torch.Tensor: r"""Sharpen an image. Args: input (torch.Tensor): Tensor to be transformed with shape :math:`(*, C, H, W)`. params (Dict[str, torch.Tensor]): - params['sharpness_factor']: Sharpness strength. Must be above 0. Returns: torch.Tensor: Adjusted image with shape :math:`(B, C, H, W)`. """ factor = params['sharpness_factor'] return sharpness(input, factor)
def apply_sharpness(input: torch.Tensor, params: Dict[str, torch.Tensor]) -> torch.Tensor: r"""Sharpen an image. Args: input (torch.Tensor): Tensor to be transformed with shape (H, W), (C, H, W), (B, C, H, W). params (Dict[str, torch.Tensor]): - params['sharpness_factor']: Sharpness strength. Must be above 0. Returns: torch.Tensor: Adjusted image. """ input = _transform_input(input) _validate_input_dtype(input, accepted_dtypes=[torch.float16, torch.float32, torch.float64]) factor = params['sharpness_factor'] return sharpness(input, factor)