Skip to content

Releases: chaofengc/IQA-PyTorch

pyiqa v0.1.11

30 Apr 09:14
Compare
Choose a tag to compare

📢 Fix Bugs

  1. 🐛 fix topiq_nr-face multiple inference error
  2. 📝 update distributed train

✨ New features

You can now use pyiqa in terminal like this:

# list all available metrics
pyiqa -ls

# test with default settings
pyiqa [metric_name(s)] --target [image_path or dir] --ref [image_path or dir]
  1. ✨ add wadiqam pretrained models
  2. ✨ add qalign
  3. 🔨 add scandir_images func
  4. 🚩 add inception_score
  5. 🚩 add console entry point with pyiqa command

🛠️ Improvements

  1. add star-history
  2. 🔧 set seed for every forward in test mode

🍻 New Contributors

Many thanks to the valuable contributions 🤗 !

Full Changelog: v0.1.10...v0.1.11

pyiqa v0.1.10

06 Jan 10:44
Compare
Choose a tag to compare

📢 Fix Bugs

  1. fix vertical flip in data transforms 872e6ca
  2. fix bug of fid calculation for different sizes 2da824d
  3. fix maniqa hyperparameter error 33a2829
  4. fix device handling 6b1547f

✨ New features

  1. Add huggingface link to download datasets 031fafd
  2. Add liqe liqe_mix by @zwx8981
  3. Add nima-koniq and nima-spaq bd385c7

🛠️ Improvements

  1. Improve training pipeline 43fb392
  2. Add more error messages in fid c36d31c
  3. Improve error message for FR and demo codes 921bd75

🍻 New Contributors

  • Improve SSIM numerical stability by transforming operations into float64 by @Luciennnnnnn in #125
  • Add LIQE by @zwx8981 in #127
    Many thanks to their valuable contributions 🤗 !

Full Changelog: v0.1.8...v0.1.10

pyiqa v0.1.8

09 Oct 06:57
Compare
Choose a tag to compare

📢 Fix Bugs

  1. Fix lower_better for FID be5d8c7
  2. Change mad option test_y_channel to be false by default cf84ffb
  3. Set read rgb to true in inference model to force read RGB image ad93cd4

✨ New features

  1. Add topiq_fr, topiq_nr, topiq_nr-face, topiq_iaa etc. introduced by our paper TOPIQ
  2. Add st-lpips by @abhijay9 in #93. Many thanks ❤️ !
  3. Add laion_aes introduced by LAION-Aesthetics_Predictor
  4. Add datasets PIQ2023, GFIQA f46faae

🛠️ Improvements

  1. Add documents: https://iqa-pytorch.readthedocs.io
  2. Update to torchvision>=0.13, torch>=1.12
  3. Improve dataset api 6fbae36
  4. Meta information files for training can be download automatically now.
  5. Update evaluation protocol and results 2770a7e
  6. Update results of maniqa 7ee5ea6
  7. Add assertion for brisque to force gray input 4e2c707

New Contributors

Full Changelog: v0.1.7...v0.1.8

pyiqa v0.1.7

23 Jun 12:35
Compare
Choose a tag to compare

📢 Important Changes & Bug Fix

  1. Fix verbose option in FID 49b2297
  2. Fix grad backpropagation for as_loss=True e027618
  3. Fix niqe with gray scale input 212ecef

✨ New features

  1. Add metric uranker b142d1c
  2. Add metric maniqa-koniq, maniqa-kadid fe95923
  3. Add metric clipscore for image-caption matching ecb3e5e
  4. Add metric entropy to calculate gray scale image entropy like matlab 5f6d4fb
  5. Add pytest cases for results calibration, datasets loading and gradient backward. bc5e135

🛠️ Improvements

  1. Recursively find images in folder for FID calculation d7ade54
  2. Add metric output doc in model cards

Hotfix of NRQM & PI

29 Mar 14:09
Compare
Choose a tag to compare

🚨 Hotfix: NRQM & PI Calculation Bug Resolved

We have identified and resolved a critical bug in the NRQM calculation within our toolbox. The issue stemmed from the SSIM (Structural Similarity Index) function, where only the structure similarity score was being utilized. We apologize for any inconvenience this may have caused.

As a result of this bug, the PI, which is determined by the formula PI = 0.5 * (10 - NRQM + NIQE), was also affected.

With this hotfix, we have:

  • Updated the NRQM calculation to correctly incorporate all relevant components of the SSIM function.
  • Adjusted the PI calculation to reflect the corrected NRQM values.

We strongly recommend using the latest release to benefit from these crucial fixes.

If you encounter any issues or have further questions, please don't hesitate to reach out to our support team. Thank you for your understanding and continued support.

IQA-PyTorch v0.1.6

20 Mar 05:43
Compare
Choose a tag to compare

⚠️ Important Changes & Bug Fix

  1. Fix OOM on GPU for NRQM fcb7f6e
  2. Fix device problem for gradient calculation 08f8850
  3. Fix bug for small image test of NIQE 8d7462d
  4. Fix default dataset config 1baa70e
  5. Fix clip installation error 53d176f
  6. feat: add psnry for y colorspace; ssimc for RGB ssim

New features

  1. Add CNNIQA, TreS
  2. Pass forward argument to inference model
  3. Add loss reduction when using metric as loss f03d7f1

Improvements

  1. Update benchmark results
  2. Update clipiqa+

IQA-PyTorch v0.1.5

26 Oct 08:53
Compare
Choose a tag to compare

⚠️ Fix bugs

  1. Fix FID bug
  2. Fix read meta info error in livechallenge.
  3. Fix shape error for NRQM
  4. Fix bug in nancov
  5. Add missing requirements package
  6. Fix link for lpips squeeze net version

New features

  1. Add MANIQA, AHIQ pretrained weights
  2. Add metric_mode option for list_models
  3. Add new metrics: FID, MANIQA
  4. Enable image path as inputs. See demo codes in README
  5. Add as_loss option to enable gradient backpropagation for metric. Default False.

Improvements

  1. Use epoch instead of iteration in lr scheduler
  2. Add clean_state_dict before loading pretrain model

IQA-PyTorch v0.1.4

20 Jun 08:02
Compare
Choose a tag to compare

New features

  1. Add new metrics: FID, MANIQA
  2. Enable image path as inputs. See demo codes in README
  3. Add as_loss option to enable gradient backpropagation for metric. Default False.

Fix bugs

  1. Fix rmse error
  2. Fix benchmark test with PieAPP

Improvements

  1. Disable gradient calculation by default for convenience.
  2. Add filter2 function to matlab utils
  3. Add reduction option to EMDLoss
  4. Add crop_border option to PSNR, SSIM

pyiqa v0.1.3 beta version

01 May 10:35
Compare
Choose a tag to compare

New features

  1. Add RMSE metric
  2. Add scale fitting option for calculation of PLCC and RMSE

Fix bugs

  1. Fix NIQE error when calculating images with large (>96 x 96) plain regions (regions with constant value). See #23
  2. Correct batch inference error for pieapp
  3. Fix compatibility of "torch.linalg.svd" for pytorch 1.9 #25

Improvements

  1. Improve function interface to match original matlab codes, including nanmean, nancov, blockproc, fspecial.
  2. Improve efficiency of symmetric padding, according to this link
  3. For pieapp, we change default stride to 27 for computation-performance trade off.

IQA-PyTorch v0.1.3 Alpha version

19 Mar 13:51
Compare
Choose a tag to compare

New features

  1. We add the following new metrics:
    • pieapp
    • paq2piq
    • dbcnn trained with our own splits and configurations
  2. Add SRCC based loss function

Important change

We change the default musiq weights from musiq-ava to musiq-koniq because it is more robust according to NR benchmark results

Fix bugs

  • Remove Lambda transform in dataset to enable distributed training
  • Fix paq2piq batch test error