Image single distortion type judgment method based on two-channel convolutional neural network

doi:10.11772/j.issn.1001-9081.2020091362

Journal of Computer Applications ›› 2021, Vol. 41 ›› Issue (6): 1761-1766.DOI: 10.11772/j.issn.1001-9081.2020091362

Special Issue: 多媒体计算与计算机仿真

• Multimedia computing and computer simulation • Previous Articles Next Articles

Image single distortion type judgment method based on two-channel convolutional neural network

YAN Junhua^1,2, HOU Ping^1,2, ZHANG Yin^1,2, LYU Xiangyang^1,2, MA Yue^1,2, WANG Gaofei^1,2

1. Key Laboratory of Space Photoelectric Detection and Perception, Ministry of Industry and Information Technology(Nanjing University of Aeronautics and Astronautics), Nanjing Jiangsu 211106, China;
2. College of Astronautics, Nanjing University of Aeronautics and Astronautics, Nanjing Jiangsu 211106, China

Received:2020-09-04 Revised:2020-11-25 Online:2020-12-07 Published:2021-06-10
Supported by:
This work is partially supported by the National Natural Science Foundation of China (61705104), the Fundamental Research Funds for the Central Universities (NJ2020021), the Natural Science Foundation of Jiangsu Province (BK20170804).

基于双通道卷积神经网络的图像单失真类型判定方法

闫钧华^1,2, 侯平^1,2, 张寅^1,2, 吕向阳^1,2, 马越^1,2, 王高飞^1,2

1. 空间光电探测与感知工业和信息化部重点实验室(南京航空航天大学), 南京 211106;
2. 南京航空航天大学航天学院, 南京 211106

通讯作者: 闫钧华
作者简介:闫钧华(1972-),女,陕西兴平人,教授,博士,主要研究方向:图像质量评价、多源信息融合、目标检测、跟踪与识别;侯平(1993-),男,江苏靖江人,硕士研究生,主要研究方向:图像质量评价;张寅(1986-),男,江苏镇江人,讲师,博士,主要研究方向:光电探测信息仿真与处理;吕向阳(1996-),男,河北景县人,硕士研究生,主要研究方向:图像质量评价;马越(1991-),男,江苏淮安人,硕士研究生,主要研究方向:图像质量评价;王高飞(1996-),女,河南新野人,硕士研究生,主要研究方向:图像处理。
基金资助:
国家自然科学基金资助项目（61705104）；中央高校基本科研业务费专项资金资助项目（NJ2020021）；江苏省自然科学基金资助项目（BK20170804）。

Abstract

Abstract: In order to solve the problem of low accuracy of some distortion types judgment by image single distortion type judgment algorithm, an image single distortion type judgment method based on two-channel Convolutional Neural Network (CNN) was proposed. Firstly, the fixed size image block was obtained by cropping the image, and the high-frequency information map was obtained by Haar wavelet transform of the image block. Then, the image block and the corresponding high-frequency information map were respectively input into the convolutional layers of different channels to extract the deep feature map, and the deep features were fused and input into the fully connected layer. Finally, the values of the last layer of the fully connected layer were input into the Softmax function classifier to obtain the probability distribution of the single distortion type of the image. Experimental results on LIVE database show that, the proposed method has the image single distortion type judgement accuracy up to 95.21%, and compared with five other image single distortion type judgment methods for comparison, the proposed method has the accuracies for judging JPEG2000 and fast fading distortions improved by at least 6.69 percentage points and 2.46 percentage points respectively. The proposed method can accurately identify the single distortion type in the image.

Key words: single distortion type, Convolutional Neural Network (CNN), wavelet transform, two-channel, high-frequency information map

摘要： 针对图像单失真类型判定算法对部分失真类型判定精度低的问题，提出了一种基于双通道卷积神经网络（CNN）的图像单失真类型判定方法。首先，对图像进行裁剪以得到固定尺寸的图像块，并对图像块进行Haar小波变换从而得到高频信息图；然后，将图像块与对应的高频信息图分别输入到不同通道卷积层中以提取深层特征图后，对深层特征进行融合并输入到全连接层中；最后，将全连接层最后一层的值输入到Softmax函数分类器中得到图像单失真类型概率分布。LIVE数据库上的实验结果表明，所提方法的图像单失真类型判定准确率达到了95.21%，并且对JPEG2000和快速衰落失真这两种失真类型的判定精度相较用于对比的其他五种图像单失真类型判定方法分别提升了至少6.69个百分点和2.46个百分点。所提方法能够准确地判定出图像中存在的单失真类型。

关键词: 单失真类型, 卷积神经网络, 小波变换, 双通道, 高频信息图

CLC Number:

TP391.41

YAN Junhua, HOU Ping, ZHANG Yin, LYU Xiangyang, MA Yue, WANG Gaofei. Image single distortion type judgment method based on two-channel convolutional neural network[J]. Journal of Computer Applications, 2021, 41(6): 1761-1766.

闫钧华, 侯平, 张寅, 吕向阳, 马越, 王高飞. 基于双通道卷积神经网络的图像单失真类型判定方法[J]. 计算机应用, 2021, 41(6): 1761-1766.

References

[1] MOORTHY A K, BOVIK A C A two-step framework for constructing blind image quality indices[J]. IEEE Signal Processing Letters,2010,17(5):513-516.
[2] MOORTHY A K, BOVIK A C. Statistics of natural image distortions[C]//Proceedings of the 2010 IEEE International Conference on Acoustics, Speech and Signal Processing. Piscataway:IEEE,2010:962-965.
[3] MOORTHY A K,BOVIK A C. Blind image quality assessment:from natural scene statistics to perceptual quality[J]. IEEE Transactions on Image Processing,2011,20(12):3350-3364.
[4] CHETOUANI A,BEGHDADI A,DERICHE M. A hybrid system for distortion classification and image quality evaluation[J]. Signal Processing:Image Communication,2012,27(9):948-960.
[5] MITTAL A,MOORTHY A K,BOVIK A C. No-reference image quality assessment in the spatial domain[J]. IEEE Transactions on Image Processing,2012,21(12):4695-4708.
[6] KANG L,YE P,LI Y,et al. Simultaneous estimation of image quality and distortion via multi-task convolutional neural networks[C]//Proceedings of the 2015 IEEE International Conference on Image Processing. Piscataway:IEEE,2015:2791-2795.
[7] WANG H,ZUO L,FU J. Distortion recognition for image quality assessment with convolutional neural network[C]//Proceedings of the 2016 IEEE International Conference on Multimedia and Expo. Piscataway:IEEE,2016:1-6.
[8] 邬美银, 陈黎, 田菁. 基于卷积神经网络的视频图像失真检测及分类[J]. 计算机应用研究, 2016, 33(9):2827-2830.(WU M Y, CHEN L, TIAN J. Video image distortion detection and classification based on CNN[J]. Application Research of Computers,2016,33(9):2827-2830.)
[9] MA K,LIU W,ZHANG K,et al. End-to-end blind image quality assessment using deep neural networks[J]. IEEE Transactions on Image Processing,2018,27(3):1202-1213.
[10] LIANG D,GAO X,LU W,et al. Deep multi-label learning for image distortion identification[J]. Signal Processing,2020,172:Article No. 107536.
[11] 朱可. 面向退化增强及复原图像的主客观一致图像质量评价方法[D]. 南京:南京航空航天大学, 2017:17-20.(ZHU K. Image quality assessment satisfying subjective and objective consistency for degraded, enhanced and restored images[D]. Nanjing:Nanjing University of Aeronautics and Astronautics, 2017:17-20.)
[12] SZEGEDY C,VANHOUCKE V,IOFFE S,et al. Rethinking the inception architecture for computer vision[C]//Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition. Piscataway:IEEE,2016:2818-2826.
[13] SIMONYAN K, ZISSERMAN A. Very deep convolutional networks for large-scale image recognition[EB/OL].[2020-08-10]. https://arxiv.org/pdf/1409.1556.pdf.
[14] NAIR V,HINTON G E. Rectified linear units improve restricted Boltzmann machines[C]//Proceedings of the 2010 27th International Conference on Machine Learning. Madison:Omnipress,2010:807-814.
[15] KIM J,LEE S. Fully deep blind image quality predictor[J]. IEEE Journal of Selected Topics in Signal Processing,2017,11(1):206-220.
[16] SHEIKH H R,WANG Z,CORMACK L,et al. Live image quality assessment database release 2[DB/OL].[2020-08-10]. http://live.ece.utexas.edu/research/quality.
[17] 肖永旗. 基于自然场景统计的无参考遥感图像质量评价[D]. 南京:南京航空航天大学, 2018:11-36.(XIAO Y Q. Noreference remote sensing image quality assessment method based on natural scene statistics[D]. Nanjing:Nanjing University of Aeronautics and Astronautics,2018:11-36.)

Image single distortion type judgment method based on two-channel convolutional neural network

基于双通道卷积神经网络的图像单失真类型判定方法

PDF

Knowledge

Abstract

Cite this article

share this article

References

Related Articles 15

Recommended Articles

Metrics

[1]	Yun LI, Fuyou WANG, Peiguang JING, Su WANG, Ao XIAO. Uncertainty-based frame associated short video event detection method [J]. Journal of Computer Applications, 2024, 44(9): 2903-2910.
[2]	Hong CHEN, Bing QI, Haibo JIN, Cong WU, Li’ang ZHANG. Class-imbalanced traffic abnormal detection based on 1D-CNN and BiGRU [J]. Journal of Computer Applications, 2024, 44(8): 2493-2499.
[3]	Dongwei WANG, Baichen LIU, Zhi HAN, Yanmei WANG, Yandong TANG. Deep network compression method based on low-rank decomposition and vector quantization [J]. Journal of Computer Applications, 2024, 44(7): 1987-1994.
[4]	Yangyi GAO, Tao LEI, Xiaogang DU, Suiyong LI, Yingbo WANG, Chongdan MIN. Crowd counting and locating method based on pixel distance map and four-dimensional dynamic convolutional network [J]. Journal of Computer Applications, 2024, 44(7): 2233-2242.
[5]	Mengyuan HUANG, Kan CHANG, Mingyang LING, Xinjie WEI, Tuanfa QIN. Progressive enhancement algorithm for low-light images based on layer guidance [J]. Journal of Computer Applications, 2024, 44(6): 1911-1919.
[6]	Jianjing LI, Guanfeng LI, Feizhou QIN, Weijun LI. Multi-relation approximate reasoning model based on uncertain knowledge graph embedding [J]. Journal of Computer Applications, 2024, 44(6): 1751-1759.
[7]	Mu LI, Yu LUO, Xizheng KE. Human vital signs detection algorithm based on frequency modulated continuous wave radar [J]. Journal of Computer Applications, 2024, 44(6): 1978-1986.
[8]	Min SUN, Qian CHENG, Xining DING. CBAM-CGRU-SVM based malware detection method for Android [J]. Journal of Computer Applications, 2024, 44(5): 1539-1545.
[9]	Wenshuo GAO, Xiaoyun CHEN. Point cloud classification network based on node structure [J]. Journal of Computer Applications, 2024, 44(5): 1471-1478.
[10]	Jie WANG, Hua MENG. Image classification algorithm based on overall topological structure of point cloud [J]. Journal of Computer Applications, 2024, 44(4): 1107-1113.
[11]	Tianhua CHEN, Jiaxuan ZHU, Jie YIN. Bird recognition algorithm based on attention mechanism [J]. Journal of Computer Applications, 2024, 44(4): 1114-1120.
[12]	Lijun XU, Hui LI, Zuyang LIU, Kansong CHEN, Weixuan MA. 3D-GA-Unet： MRI image segmentation algorithm for glioma based on 3D-Ghost CNN [J]. Journal of Computer Applications, 2024, 44(4): 1294-1302.
[13]	Jingxian ZHOU, Xina LI. UAV detection and recognition based on improved convolutional neural network and radio frequency fingerprint [J]. Journal of Computer Applications, 2024, 44(3): 876-882.
[14]	Ruifeng HOU, Pengcheng ZHANG, Liyuan ZHANG, Zhiguo GUI, Yi LIU, Haowen ZHANG, Shubin WANG. Iterative denoising network based on total variation regular term expansion [J]. Journal of Computer Applications, 2024, 44(3): 916-921.
[15]	Yongfeng DONG, Jiaming BAI, Liqin WANG, Xu WANG. Chinese named entity recognition combining prior knowledge and glyph features [J]. Journal of Computer Applications, 2024, 44(3): 702-708.