This is the source code for the CVPR'20 paper "Blindly Assess Image Quality in the Wild Guided by A Self-Adaptive Hyper Network".
- Python 3.6+
- PyTorch 0.4+
- TorchVision
- scipy
(optional for loading specific IQA Datasets)
- csv (KonIQ-10k Dataset)
- openpyxl (BID Dataset)
Predicting image quality with our model trained on the Koniq-10k Dataset.
To run the demo, please download the pre-trained model at Google drive or Baidu cloud (password: 1ty8), put it in 'pretrained' folder, then run:
python demo.py
You will get a quality score ranging from 0-100, and a higher value indicates better image quality.
Training and testing our model on the LIVE Challenge Dataset.
python train_test_IQA.py
Some available options:
--dataset: Training and testing dataset, support datasets: livec | koniq-10k | bid | live | csiq | tid2013.--train_patch_num: Sampled image patch number per training image.--test_patch_num: Sampled image patch number per testing image.--batch_size: Batch size.
When training or testing on CSIQ dataset, please put 'csiq_label.txt' in your own CSIQ folder.
If you find this work useful for your research, please cite our paper:
@InProceedings{Su_2020_CVPR,
author = {Su, Shaolin and Yan, Qingsen and Zhu, Yu and Zhang, Cheng and Ge, Xin and Sun, Jinqiu and Zhang, Yanning},
title = {Blindly Assess Image Quality in the Wild Guided by a Self-Adaptive Hyper Network},
booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
month = {June},
year = {2020}
}