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

• 作者: 闫钧华 侯平 张寅 吕向阳 马越 王高飞
• 来源期刊: 计算机应用
• 关键词: 单失真类型 卷积神经网络 小波变换 双通道 高频信息图
• 年，卷（期）: 2021,（6）
• 所属期刊栏目: 多媒体计算与计算机仿真|Multimedia computing and computer simulation
• 页码范围: 1761-1766
• 页数: 6页
• 分类号: TP391.41
• 语种: 中文
• DOI: 10.11772/j.issn.1001-9081.2020091362

摘要

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