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

权利要求书

1.一种去雾图像的质量评价方法，其特征在于，包括：

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

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

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

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

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

2.根据权利要求1所述的质量评价方法，其特征在于，多个特征提取块包括基本块和扩张块，所述利用预设去雾图像初始评价模型中的多个基本块和扩张块，分别对所述多个雾度相关特征进行特征提取，并将提取的特征进行特征融合，包括：

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

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

3.根据权利要求2所述的质量评价方法，其特征在于，所述基本块包括第一基本块和第二基本块，所述扩张块包括第一扩张块、第二扩张块和第三扩张块，所述利用所述基本块和所述扩张块提取每个雾度相关特征并分别输出不同的尺度特征，包括：

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

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

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

将每个所述第一尺度特征进行特征拼接得到第一拼接特征，将所述第一拼接特征与每个所述第二尺度特征进行拼接得到第二拼接特征，将所述第二拼接特征与每个所述第三尺度特征进行拼接得到第三拼接特征。

4.根据权利要求1所述的质量评价方法，其特征在于，所述利用注意力模型对融合后的特征进行深度聚合得到聚合结果，包括：

利用通道注意力模块对融合后的特征进行聚合得到第一聚合特征；

利用对比度注意力模块对融合后的特征进行聚合得到第二聚合特征；

将所述第一聚合特征和所述第二聚合特征进行元素相乘得到聚合结果。

5.根据权利要求1至4任意一项所述的质量评价方法，其特征在于，排序损失函数L为：

L＝L_rank+L₁，

其中，D＝-[Q_pre(x_i)-Q_pre(x_j)][Q_gt(x_i)-Q_gt(x_j)]，N是批训练样本数量；Q_pre(x)和Q_gt(x)分别表示输入图像的预测质量分数和真实参考质量分数，x_i和x_j是批训练图像中的索引，范围分别为[1，N-1]和[2，N]，D表示预测与真实质量差值之间的距离。