pyiqa.matlab_utils.scfpyr_util¶

Complex-valued steerable pyramid

Created by: https://github.com/tomrunia/PyTorchSteerablePyramid

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

Refer to:

Official Matlab code from https://github.com/LabForComputationalVision/matlabPyrTools/blob/master/buildSCFpyr.m;
Original Python code from https://github.com/LabForComputationalVision/pyPyrTools/blob/master/pyPyrTools/SCFpyr.py;

Module Contents¶

pyiqa.matlab_utils.scfpyr_util.pointOp[source]¶

class pyiqa.matlab_utils.scfpyr_util.SCFpyr_PyTorch(height=5, nbands=4, scale_factor=2, device=None)[source]¶

Bases: object

This is a modified version of buildSFpyr, that constructs a complex-valued steerable pyramid using Hilbert-transform pairs of filters. Note that the imaginary parts will not be steerable. Pytorch version >= 1.8.0

build(im_batch)[source]¶

Decomposes a batch of images into a complex steerable pyramid. The pyramid typically has ~4 levels and 4-8 orientations.

Parameters:: im_batch (torch.Tensor) – Batch of images of shape [N,C,H,W]
Returns:: list containing torch.Tensor objects storing the pyramid
Return type:: pyramid