[发明专利]一种基于注意力模块的图像质量评价方法

摘要

本发明提出了一种基于注意力模块的图像质量评价方法。本发明将训练集中的原始图像切割成子图像并归一化，通过人工评分得到原始图像的质量分数目标值；利用双分支卷积神经网络建立回归网络模型得到原始图像的质量分数预测值，将其与目标值构建损失函数，利用损失函数对所建立的回归网络模型进行优化训练；将待评价图像经过处理后得到其多张子图像，输入训练后的回归网络模型中，最终得到待评价图像的质量评分。本发明同时满足了低层数和深层效果；引入混合注意力模块嵌入，使得网络在空间和通道上的抗噪声干扰能力均有提高；利用两个网络提取多尺度图像特征，并采用不同融合方式对于网络输出做混合处理，保留关键特征的同时减少参数冗余。