Metadata-Version: 2.1
Name: IQA-pytorch
Version: 0.1
Summary: IQA models in PyTorch
Home-page: https://github.com/dingkeyan93/IQA-pytorch
Author: Keyan Ding
Author-email: dingkeyan93@outlook.com
License: MIT
Description: # Image Quality Assessment (IQA) Models in PyTorch
        
        This is a repository to re-implement the existing IQA models with PyTorch, including
        - [SSIM](https://www.cns.nyu.edu/~lcv/ssim/), [MS-SSIM](https://ece.uwaterloo.ca/~z70wang/publications/msssim.html), [CW-SSIM](https://www.mathworks.com/matlabcentral/fileexchange/43017-complex-wavelet-structural-similarity-index-cw-ssim),
        - [FSIM](https://sse.tongji.edu.cn/linzhang/IQA/FSIM/FSIM.htm), [VSI](https://sse.tongji.edu.cn/linzhang/IQA/VSI/VSI.htm),[GMSD](https://www4.comp.polyu.edu.hk/~cslzhang/IQA/GMSD/GMSD.htm),
        - [NLPD](https://www.cns.nyu.edu/~lcv/NLPyr/), [MAD](http://vision.eng.shizuoka.ac.jp/mod/url/view.php?id=54),
        - [VIF](https://live.ece.utexas.edu/research/Quality/VIF.htm),
        - [LPIPS](https://github.com/richzhang/PerceptualSimilarity), [DISTS](https://github.com/dingkeyan93/DISTS).
        
        **Note:** The reproduced results may be a little different from the original matlab version.
        
        #### Installation:
        - ```pip install IQA_pytorch```
        
        #### Requirements: 
        - Python>=3.6
        - Pytorch>=1.2
        
        #### Usage:
        ```python
        from IQA_pytorch import SSIM, GMSD, LPIPSvgg, DISTS
        D = SSIM()
        # Calculate score of the image X with the reference Y
        # X: (N,3,H,W) 
        # Y: (N,3,H,W) 
        # Tensor, data range: 0~1
        score = D(X, Y, as_loss=False) 
        # set 'as_loss=True' to get a value as loss for optimizations.
        loss = D(X, Y)
        loss.backward()
        ```
        
        <!-- ### Citation
        ```
        @article{ding2020iqa,
          title={Image Quality Assessment: Unifying Structure and Texture Similarity},
          author={Ding, Keyan and Ma, Kede and Wang, Shiqi and Simoncelli, Eero P.},
          journal = {CoRR},
          volume = {abs/2004.07728},
          year={2020},
          url = {https://arxiv.org/abs/2004.07728}
        }
        ``` -->
Keywords: pytorch,similarity,IQA,metric,image-quality
Platform: python
Description-Content-Type: text/markdown
