/deepIQA

NR and FR IQA models based on Deep Convolutional Neural Networks

Primary LanguagePythonMIT LicenseMIT

deepIQA

This is the reference implementation of Deep Neural Networks for No-Reference and Full-Reference Image Quality Assessment. The pretrained models contained in the models directory were trained for both NR and FR IQA and for both model variants described in the paper. They were trained on the full LIVE or TID2013 database respectively, as used in the cross-dataset evaluations. This evaluation script uses non-overlapping 32x32 patches to produce deterministic scores, whereas the evaluation in the paper uses randomly sampled overlapping patches.

usage: evaluate.py [-h] [--model MODEL] [--top {patchwise,weighted}] [--gpu GPU] INPUT [REF]

Dependencies

TODO

  • add training code
  • add cpu support (minor change)
  • remove opencv and scikit-learn dependencies for loading data (minor changes)
  • fix non-deterministic behaviour