pyiqa.archs.msswd_arch

Perceptual color difference metric, MS-SWD.

@inproceedings{he2024ms-swd,

title={Multiscale Sliced {Wasserstein} Distances as Perceptual Color Difference Measures}, author={He, Jiaqi and Wang, Zhihua and Wang, Leon and Liu, Tsein-I and Fang, Yuming and Sun, Qilin and Ma, Kede}, booktitle={European Conference on Computer Vision}, pages={1–18}, year={2024}, url={http://arxiv.org/abs/2407.10181}

}

Reference:

• Official github: https://github.com/real-hjq/MS-SWD

Module Contents

pyiqa.archs.msswd_arch.color_space_transform(input_color, fromSpace2toSpace)

Transforms inputs between different color spaces :param input_color: tensor of colors to transform (with NxCxHxW layout) :param fromSpace2toSpace: string describing transform :return: transformed tensor (with NxCxHxW layout)

class pyiqa.archs.msswd_arch.MS_SWD_learned(resize_input: bool = True, pretrained: bool = True, pretrained_model_path: str = None, **kwargs)

Bases: torch.nn.Module

preprocess_img(x)

forward_once(x)

forward(x, y)