pyiqa.matlab_utils.nss_feature ============================== .. py:module:: pyiqa.matlab_utils.nss_feature Module Contents --------------- .. py:function:: estimate_aggd_param(block: torch.Tensor, return_sigma=False) -> Tuple[torch.Tensor, torch.Tensor, torch.Tensor] 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). :rtype: Tensor .. py:function:: compute_nss_features(luma_nrmlzd: torch.Tensor) -> torch.Tensor