The PyTorch implementation of DB-CNN is released at https://github.com/zwx8981/DBCNN-PyTorch!
Weixia Zhang, Kede Ma, Jia Yan, Dexiang Deng, and Zhou Wang https://ieeexplore.ieee.org/document/8576582
IEEE Transactions on Circuits and Systems for Video Technology (TCSVT), Volume: 30 , Issue: 1 , Jan. 2020.
Usuage:
distorted_img = distortion_generator( img, dist_type, level, seed )
Where img is the original pristine image, dist_type refers to a specified distortion type ranging in 1~9.
1, Gaussian Blur
2, White Noise
3, JPEG Compression
4, JPEG2000 Compression
5, Contrast Change
6, Pink Noise
7, Image Color Quantization with Dither
8, Over-Exposure
9, Under-Exposure
level is a specified degradation level range in 1~5.
seed should be fixed to be 1.
Running the run_exp.m script to train and test on a specifid dataset across 10 random splits.
Prerequisite: Matlab(We use 2017a), MatConvNet (We use 1.0-beta25)， vlfeat(We use 0.9.2)
Pretrained s-cnn model is included in dbcnn\data\models, you should download vgg-16 model from http://www.vlfeat.org/matconvnet/pretrained/ and put it in dbcnn\data\models.
You need to copy the matconvet/matlab folder to that of your matconvnet to modify the vl_simplenn.m and PDist.m files.
Relevant links:
Waterloo Exploration Database: https://ece.uwaterloo.ca/~k29ma/exploration/
PASCAL VOC 2012: http://host.robots.ox.ac.uk/pascal/VOC/voc2012/
@article{zhang2020blind,
title={Blind Image Quality Assessment Using A Deep Bilinear Convolutional Neural Network},
author={Zhang, Weixia and Ma, Kede and Yan, Jia and Deng, Dexiang and Wang, Zhou},
journal={IEEE Transactions on Circuits and Systems for Video Technology},
volume={30},
number={1},
pages={36--47},
year={2020}
}