[发明专利]图像处理模型的训练方法、电子设备及计算机存储介质

说明书

本申请实施例提供了一种图像处理模型的训练方法、电子设备及计算机存储介质，图像处理模型的训练方法，包括：获得样本图像组，样本图像组包括图像内容相同的第一图像和第二图像；通过图像处理模型，根据图像特征及其对应的预设区域划分参数对第一图像进行区域划分，得到至少两类待模糊区域；根据各类待模糊区域的图像特征及其分别对应的去模糊参数，构造各类待模糊区域分别对应的模糊程度数据，得到第一图像对应的模糊程度数据；通过图像处理模型根据第一图像对应的模糊程度数据对第二图像进行去模糊处理，得到去模糊图像，根据第一图像和去模糊图像计算损失值，根据损失值调整图像处理模型，以通过调整后的图像处理模型对图像进行去模糊处理。

技术领域

本申请实施例涉及视频图像处理领域，尤其涉及一种图像处理模型的训练方法、电子设备及计算机存储介质。

背景技术

会议室音视频采集、线上直播、实时会议、实时通讯(real time communication，RTC)等场景，受到镜头、视频采集环境的限制，以及下采样和编解码压缩的影响，画质容易趋于模糊，并伴随各种失真/瑕疵(artifacts)。

一般情况下，可以采用去模糊(deblur)、提成分辨率(超分)、去瑕疵等方法提升画质，上述三种中去模糊的方式对画质的提升效果较好。通常采用的去模糊方案需要根据图像内容估计对应的模糊核，并根据模糊核进行去模糊，计算成本高且效果差。

发明内容

有鉴于此，本申请实施例提供一种图像处理模型的训练方案，以至少部分解决上述问题。

根据本申请实施例的第一方面，提供了一种图像处理模型的训练方法，包括：获得样本图像组，所述样本图像组包括图像内容相同的第一图像和第二图像，第一图像的清晰度大于第二图像的清晰度；通过图像处理模型对所述第一图像进行图像特征提取，得到图像特征；通过所述图像处理模型，根据所述图像特征及其对应的预设区域划分参数对所述第一图像进行区域划分，得到至少两类待模糊区域；根据各类待模糊区域的图像特征及其分别对应的去模糊参数，构造各类待模糊区域分别对应的模糊程度数据，得到所述第一图像对应的模糊程度数据；通过所述图像处理模型根据所述第一图像对应的模糊程度数据对所述第二图像进行去模糊处理，得到去模糊图像，根据所述第一图像和所述去模糊图像计算损失值，根据损失值调整所述图像处理模型中的以下至少之一：预设区域划分参数、去模糊参数，以通过调整后的所述图像处理模型对图像进行去模糊处理。

根据本申请实施例的第二方面，提供了一种图像处理模型的部署方法，包括：发送待处理图像，以通过预先训练好的多个图像处理模型对所述待处理图像进行图像特征提取，得到图像特征；以及通过所述图像处理模型，根据所述图像特征与预设区域划分参数的匹配程度对所述待处理图像进行区域划分，得到至少两类待处理区域；并通过所述图像处理模型，根据各类待处理区域分别对应的去模糊参数，将至少两类处理区域分别进行去模糊处理，得到多个图像处理模型分别输出的去模糊图像，其中，所述多个图像处理模型的图像参数存在差异，所述图像参数包括预设区域划分参数和\或去模糊参数；向用户展示多个所述去模糊图像；发送用户从所述多个去模糊图像中选择的一个去模糊图像，以接收并部署被选择的去模糊图像对应的所述图像处理模型，通过所述图像处理模型进行去模糊处理。

根据本申请实施例的第三方面，提供了一种图像处理方法，包括：确定待处理图像；通过图像处理模型对所述待处理图像进行图像特征提取，得到图像特征；通过所述图像处理模型，根据所述图像特征与预设区域划分参数的匹配程度对所述待处理图像进行区域划分，得到至少两类待处理区域；通过所述图像处理模型，根据各类待处理区域分别对应的去模糊参数，将至少两类处理区域分别进行去模糊处理，得到所述待处理图像对应的去模糊图像。

根据本申请实施例的第四方面，提供了一种电子设备，包括：处理器、存储器、通信接口和通信总线，所述处理器、所述存储器和所述通信接口通过所述通信总线完成相互间的通信；所述存储器用于存放至少一可执行指令，所述可执行指令使所述处理器执行如上所述方法对应的操作。