Image Quality Assessment (IQA) Models in PyTorch
This is a repository to re-implement the existing IQA models with PyTorch, including
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:
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()
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
IQA_pytorch-0.1.tar.gz
(38.9 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
IQA_pytorch-0.1-py3-none-any.whl
(62.6 kB
view details)
File details
Details for the file IQA_pytorch-0.1.tar.gz.
File metadata
- Download URL: IQA_pytorch-0.1.tar.gz
- Upload date:
- Size: 38.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.4.0 requests-toolbelt/0.9.1 tqdm/4.36.1 CPython/3.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
7f2064815f42361ab38f84edf435bdf505a7d9862e0ab15a8e60be2bce9c7b50
|
|
| MD5 |
e33179d7865bee32753396ac23cf75cd
|
|
| BLAKE2b-256 |
9149c8c39338c2f1c7baa581d0361c4876d3f15131fb101f8f5617c885054908
|
File details
Details for the file IQA_pytorch-0.1-py3-none-any.whl.
File metadata
- Download URL: IQA_pytorch-0.1-py3-none-any.whl
- Upload date:
- Size: 62.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/3.1.1 pkginfo/1.5.0.1 requests/2.22.0 setuptools/41.4.0 requests-toolbelt/0.9.1 tqdm/4.36.1 CPython/3.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
45b12fd2ef168ec004cc09edae5491553ea82441cb840a5776219dd5366b93f9
|
|
| MD5 |
d456f397ab7a97b9d306b5de89ada03e
|
|
| BLAKE2b-256 |
9feedc11f49757fac3704707ac9157e8646defa485a3daf05b6cfde28b3e1abd
|
