pyiqa.matlab_utils.math_util

Mathematical utilities

Created by: https://github.com/tomrunia/PyTorchSteerablePyramid/blob/master/steerable/math_utils.py

Modified by: Jiadi Mo (https://github.com/JiadiMo)

Module Contents

pyiqa.matlab_utils.math_util.abs(x)

pyiqa.matlab_utils.math_util.roll_n(X, axis, n)

pyiqa.matlab_utils.math_util.batch_fftshift2d(x)

Args: x: An complex tensor. Shape \((N, C, H, W)\). Pytroch version >= 1.8.0

pyiqa.matlab_utils.math_util.batch_ifftshift2d(x)

Parameters:

x – An input tensor. Shape \((N, C, H, W, 2)\).

Returns:

An complex tensor. Shape \((N, C, H, W)\).

pyiqa.matlab_utils.math_util.prepare_grid(m, n)

pyiqa.matlab_utils.math_util.rcosFn(width, position)

pyiqa.matlab_utils.math_util.pointOp(im, Y, X)

pyiqa.matlab_utils.math_util.getlist(coeff)