• Stars
    star
    203
  • Rank 192,317 (Top 4 %)
  • Language
    Python
  • License
    MIT License
  • Created almost 8 years ago
  • Updated over 5 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

NR and FR IQA models based on Deep Convolutional Neural Networks

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