pyiqa.matlab_utils.nss_feature

Module Contents

pyiqa.matlab_utils.nss_feature.estimate_aggd_param(block: torch.Tensor, return_sigma=False) Tuple[torch.Tensor, torch.Tensor, torch.Tensor][source]

Estimate AGGD (Asymmetric Generalized Gaussian Distribution) parameters. :param block: Image block with shape (b, 1, h, w). :type block: Tensor

Returns:

alpha, beta_l and beta_r for the AGGD distribution (Estimating the parames in Equation 7 in the paper).

Return type:

Tensor

pyiqa.matlab_utils.nss_feature.compute_nss_features(luma_nrmlzd: torch.Tensor) torch.Tensor[source]