【技术实现步骤摘要】
图像处理模型的训练方法、电子设备及计算机存储介质
[0001]本申请实施例涉及视频图像处理领域,尤其涉及一种图像处理模型的训练方法、电子设备及计算机存储介质。
技术介绍
[0002]会议室音视频采集、线上直播、实时会议、实时通讯(real time communication,RTC)等场景,受到镜头、视频采集环境的限制,以及下采样和编解码压缩的影响,画质容易趋于模糊,并伴随各种失真/瑕疵(artifacts)。
[0003]一般情况下,可以采用去模糊(deblur)、提成分辨率(超分)、去瑕疵等方法提升画质,上述三种中去模糊的方式对画质的提升效果较好。通常采用的去模糊方案需要根据图像内容估计对应的模糊核,并根据模糊核进行去模糊,计算成本高且效果差。
技术实现思路
[0004]有鉴于此,本申请实施例提供一种图像处理模型的训练方案,以至少部分解决上述问题。
[0005]根据本申请实施例的第一方面,提供了一种图像处理模型的训练方法,包括:获得样本图像组,所述样本图像组包括图像内容相同的第一图像和第二图像,第一图像的清晰度大于第二图像的清晰度;通过图像处理模型对所述第一图像进行图像特征提取,得到图像特征;通过所述图像处理模型,根据所述图像特征及其对应的预设区域划分参数对所述第一图像进行区域划分,得到至少两类待模糊区域;根据各类待模糊区域的图像特征及其分别对应的去模糊参数,构造各类待模糊区域分别对应的模糊程度数据,得到所述第一图像对应的模糊程度数据;通过所述图像处理模型根据所述第一图像对应的模 ...
【技术保护点】
【技术特征摘要】
1.一种图像处理模型的训练方法,包括:获得样本图像组,所述样本图像组包括图像内容相同的第一图像和第二图像,第一图像的清晰度大于第二图像的清晰度;通过图像处理模型对所述第一图像进行图像特征提取,得到图像特征;通过所述图像处理模型,根据所述图像特征及其对应的预设区域划分参数对所述第一图像进行区域划分,得到至少两类待模糊区域;根据各类待模糊区域的图像特征及其分别对应的去模糊参数,构造各类待模糊区域分别对应的模糊程度数据,得到所述第一图像对应的模糊程度数据;通过所述图像处理模型根据所述第一图像对应的模糊程度数据对所述第二图像进行去模糊处理,得到去模糊图像,根据所述第一图像和所述去模糊图像计算损失值,根据损失值调整所述图像处理模型中的以下至少之一:预设区域划分参数、去模糊参数,以通过调整后的所述图像处理模型对图像进行去模糊处理。2.根据权利要求1所述的方法,其中,若所述图像特征包括边缘特征,所述预设区域划分参数包括图像边缘划分阈值,所述通过所述图像处理模型,根据所述图像特征及其对应的预设区域划分参数对所述第一图像进行区域划分,得到至少两类待模糊区域,包括:通过所述图像处理模型,根据所述边缘特征和所述图像边缘划分阈值进行图像划分,得到与图像边缘对应的第一区域;将除所述第一区域之外的区域划分为第二区域。3.根据权利要求2所述的方法,其中,所述图像特征包括纹理特征,所述边缘特征为所述纹理特征的梯度值;所述预设区域划分参数包括纹理密度阈值,所述图像边缘划分阈值为图像梯度阈值;所述通过所述图像处理模型,根据所述图像特征及其对应的预设区域划分参数对所述第一图像进行区域划分,得到至少两类待模糊区域,包括:通过所述图像处理模型,将纹理梯度大于所述纹理梯度阈值的区域划分为所述第一区域;所述方法还包括:将纹理密度大于所述纹理密度阈值的区域划分为第三区域;所述将除所述第一区域之外的区域划分为第二区域,包括:将除所述第一区域和所述第三区域之外的区域划分为所述第二区域。4.根据权利要求3所述的方法,其中,所述将纹理密度大于所述纹理密度阈值的区域划分为第三区域包括:通过预先训练的纹理密集区域识别网络识别得到所述第三区域,所述纹理密集区域识别网络通过下述方法训练:获得不包含纹理密集区域的第三图像,以及只包含纹理密集区域的第四图像;将所述第三图像和所述第四图像进行图像融合处理,得到密集样本图像,并将所述密集样本图像中属于所述第四图像的区域作为图像标注;根据所述密集样本图像和所述图像标注训练纹理密集区域识别网络。5.根据权利要求2
‑
4任一项所述的方法,其中,所述模糊程度数据包括第一模糊窗口、第二模糊窗口,所述根据各类待模糊区域的图像特征及其分别对应的去模糊参数,构造各类待模糊区域分别对应的模糊程度数据,得到所述第一图像对应的模糊程度数据,包括:
针对与图像边缘对应的所述第一区域,根据所述去模糊参数中的第一模糊窗口参数组以及所述第一区域的各个像素的图像梯度,确定所述第一区域的各个像素对应的模糊权重,根据所述第一区域的各个像素对应的模糊权重构建所述第一模糊窗口,其中,所述第一区域中图像梯度越高的像素点对应的所述模糊权重越大,其中所述模糊权重与去模糊程度成正比;针对所述第二区域,根据所述去模糊参数中的第二模糊窗口参数组以及所述第二区域的各个像素的像素值,确定所述第二区域的各个像素对应的模糊权重,根据所述第二区域的各个像素对应的模糊权重构建所述第二模糊窗口,其中,所述第二区域中像素值越高的像素点对应的所述模糊权重越小。6.根据权利要求5所述的方法,其中,若存在所述第三区域,所述模糊程度数据还包括卷积窗,所述根据各类待模糊区域的图像特征及其分别对应的去模糊参数,构造各类待模糊区域分别对应的模糊程度数据,得到所述第一图像对应的模糊程度数据,还包括:针对所述第一图像中的所述第三区域,根据所述去模糊参数中的图像增强参数以及所述第三区域的各个像素的像素值构建用于进行图像增强处理的所述卷积窗。7.根据权利要求5所述的方法,其中,所述根据各类待模糊区域的图像特征及其分别对应的去模糊参数,构造各类待模糊区域分别对应的模糊程度数据,得到所述第一图像对应的模糊程度数据之前,所述方法还包括:为所述第一图像增加模糊遮罩,其中,所述模糊遮罩用于实现以下至少之一:压低所述第一区域中与图像边缘对应的像素位置相邻的像素的模糊程...
【专利技术属性】
技术研发人员:邵纬航,刘国栋,林建良,
申请(专利权)人:阿里巴巴中国有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。