[发明专利]画质增强方法、装置、电子设备和可读存储介质

权利要求书

1.一种画质增强方法，其特征在于，包括：

获取原始待修复图像，将所述原始待修复图像的分辨率调整至预设尺寸，得到调整后的原始待修复图像；

对调整后的原始待修复图像进行低分辨率修复，得到待增强图像；

基于预先训练完成的画质增强模型，对所述待增强图像进行画质增强处理，获得画质增强后的结果图像；

所述画质增强模型通过以下步骤训练获得：获取多张高分辨率图像，对各所述高分辨率图像进行预处理，获得多张待修复样本图像；对各所述待修复样本图像进行低分辨率修复，得到多张待增强样本图像；

利用多张所述高分辨率图像、多张所述待修复样本图像及多张所述待增强样本图像对预先构建的包括画质增强子网络及超分辨率重建子网络的卷积神经网络进行训练，包括：将多张所述高分辨率图像、多张所述待修复样本图像及多张所述待增强样本图像输入所述画质增强子网络；利用多张所述高分辨率图像、多张所述待修复样本图像，基于所述画质增强子网络对各所述待增强样本图像进行画质增强，得到多个第一输出图像；基于所述超分辨率重建子网络对各所述第一输出图像进行超分辨率重建，得到多个第二输出图像；根据所述第二输出图像和所述高分辨率图像，基于预设损失函数，通过反向传播算法对所述画质增强子网络及超分辨率重建子网络的权值进行调整，直至所述预设损失函数的输出小于预设阈值，得到所述画质增强模型。

2.根据权利要求1所述的画质增强方法，其特征在于，所述对各所述高分辨率图像进行预处理，获得多张待修复样本图像的步骤包括：

按预设插值算法，对各所述高分辨率图像进行下采样，得到多张低分辨率样本图像；

对多张所述低分辨率样本图像进行降画质处理，获得多张待修复样本图像。

3.根据权利要求1所述的画质增强方法，其特征在于，所述对各所述待修复样本图像进行低分辨率修复，得到多张待增强样本图像的步骤包括：

将各所述待修复样本图像的分辨率调整至所述预设尺寸，得到调整后的多张待修复样本图像；

对调整后的多张待修复样本图像进行低分辨率修复，得到多张待增强样本图像。

4.根据权利要求3所述的画质增强方法，其特征在于，所述对调整后的多张待修复样本图像进行低分辨率修复，得到多张待增强样本图像的步骤包括：

基于预先训练好的低分辨率修复模型，对调整后的各待修复样本图像进行低分辨率修复，得到多张待增强样本图像。

5.根据权利要求1所述的画质增强方法，其特征在于，所述画质增强子网络包括深度引导滤波层及多个残差块，所述利用多张所述高分辨率图像、多张所述待修复样本图像，基于所述画质增强子网络对各所述待增强样本图像进行画质增强，得到多个第一输出图像的步骤包括：

利用多张所述高分辨率图像、多张所述待修复样本图像，基于所述深度引导滤波层对所述待增强样本图像进行画质重构处理，得到多个画质重构样本图像；

按预设的通道合并算法，合并各所述画质重构样本图像与所述高分辨率图像的处理通道；

通过所述多个残差块对合并处理通道后的所述画质重构样本图像及所述高分辨率图像进行逆退化处理，得到多个第一输出图像。