[发明专利]一种去雾图像的质量评价方法和装置

说明书

本发明实施例公开了一种去雾图像的质量评价方法，其特征在于，包括：获取待训练的去雾样本图像，从所述去雾样本图像中提取多个雾度相关特征；利用预设去雾图像初始评价模型中的多个特征提取块分别对所述多个雾度相关特征进行特征提取，并将提取的特征进行特征融合；利用注意力模型对融合后的特征进行深度聚合得到聚合结果，通过基本块和全连接层对所述聚合结果进行处理输出所述去雾样本图像的评价结果；利用预设的排序损失函数对所述评价结果进行反向训练直至所述预设去雾图像初始评价模型收敛，得到去雾图像的质量评价模型；获取待评价的去雾图像，将所述去雾图像输入至所述去雾图像的质量评价模型，得到所述去雾图像的质量评价分数。

技术领域

本发明实施例涉及计算机技术领域，尤其是一种去雾图像的质量评价方法和装置。

背景技术

图像质量评价(Image Quality Assessment,IQA)是指对图像进行特性分析研究，如清晰度、真实性等，从而评估图像质量的好坏。其在图像恢复、图像压缩、视频编解码、自动驾驶等领域均具有广泛的实用性。随着去雾算法的不断涌现，如何正确评价去雾图像的视觉质量愈发紧迫和重要。

目前，研究去雾图像质量评价主要存在以下难点：1)局限性：主观评价最精确可靠但是耗时耗力，无法进行实时应用，而在客观评价中，由于现实生活中清晰参考图像的缺失，全参考(Full-Reference,FR)和半参考(Reduced-Reference,RR)图像质量评价方法受到限制；2)针对性：近年来，一般图像失真类型(压缩、模糊等)的质量评价方法已经取得了很大的成功，而雾度图像与一般失真之间存在差异，随着人们对图像去雾的需求日益增长，而针对去雾图像的质量评价指标屈指可数，且仍缺少能够正确判别图像质量好坏的方法。

发明内容

为解决上述技术问题，本发明创造的实施例提供一种去雾图像的质量评价方法，包括：

获取待训练的去雾样本图像，从所述去雾样本图像中提取多个雾度相关特征；

利用预设去雾图像初始评价模型中的多个特征提取块分别对所述多个雾度相关特征进行特征提取，并将提取的特征进行特征融合；

利用注意力模型对融合后的特征进行深度聚合得到聚合结果，通过基本块和全连接层对所述聚合结果进行处理输出所述去雾样本图像的评价结果；

利用预设的排序损失函数对所述评价结果进行反向训练直至所述预设去雾图像初始评价模型收敛，得到去雾图像的质量评价模型；

获取待评价的去雾图像，将所述去雾图像输入至所述去雾图像的质量评价模型，得到所述去雾图像的质量评价分数。

进一步地，多个特征提取块包括基本块和扩张块，所述利用预设去雾图像初始评价模型中的多个基本块和扩张块，分别对所述多个雾度相关特征进行特征提取，并将提取的特征进行特征融合，包括：

利用所述基本块和所述扩张块提取每个雾度相关特征并分别输出不同的尺度特征；

将所述扩张块输出的不同尺度特征进行融合。

进一步地，所述基本块包括第一基本块和第二基本块，所述扩张块包括第一扩张块、第二扩张块和第三扩张块，所述利用所述基本块和所述扩张块提取每个雾度相关特征并分别输出不同的尺度特征，包括：

利用所述第一基本块对每个雾度相关特征进行特征提取得到第一雾度特征，利用第二基本块对所述第一雾度特征进行特征提取得到第二雾度特征；

利用设置有第一扩张率参数的第一扩张块对所述第二雾度特征进行特征提取得到每个雾度相关特征的第一尺度特征；利用设置有第二扩张率参数的第二扩张块对所述第一尺度特征进行特征提取得到每个雾度相关特征的第二尺度特征，利用设置有第三扩张率参数的第三扩张块对所述第二尺度特征进行特征提取得到每个雾度相关特征的第三尺度特征；

所述将所述扩张块输出的不同尺度特征进行融合，包括：