pyiqa.archs.msswd_arch
======================

.. py:module:: pyiqa.archs.msswd_arch

.. autoapi-nested-parse::

   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
---------------

.. py:function:: 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)


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

   Bases: :py:obj:`torch.nn.Module`


   .. py:method:: preprocess_img(x)


   .. py:method:: forward_once(x)


   .. py:method:: forward(x, y)