图像处理及其模型训练方法、装置及电子设备制造方法及图纸

本申请提供了一种图像处理及其模型训练方法、装置及电子设备，涉及图像处理技术领域。其中，方法包括：获取训练样本集；训练样本集中的样本包括成对的低质量图像和高质量图像、以及低质量图像对应的边缘图像；质量的评价参数包括以下之一：分辨率、清晰度、色彩饱和度；利用训练样本集中的样本对初始网络模型进行训练，得到图像处理模型；其中，训练样本集中的边缘图像作为先验信息参与训练；图像处理模型包括图像超分辨率模型或图像增强模型。本申请能够提升图像超分辨率模型或图像增强模型图像处理的视觉效果。处理的视觉效果。处理的视觉效果。

【技术实现步骤摘要】
图像处理及其模型训练方法、装置及电子设备


[0001]本申请涉及软件图像处理
，尤其是涉及一种图像处理及其模型训练方法、装置及电子设备。

技术介绍

[0002]图像超分辨率技术是指由一幅低分辨率图像或图像序列恢复出高分辨率图像；图像增强技术是将视觉效果较差的图像处理为视觉效果更好的图像。现有技术中，通常采用一些神经网络进行图像超分辨率模型或图像增强模型的训练，而训练样本往往是大量成对的低分辨率图像和高分辨率图像，或者是成对的低清晰度图像和高清晰度图像；通过这些训练样本训练出的模型，其图像处理效果一般，无法达到更好的视觉效果。

技术实现思路

[0003]本申请的目的在于提供一种图像处理及其模型训练方法、装置及电子设备，能够提升图像超分辨率模型或图像增强模型图像处理的视觉效果。
[0004]第一方面，本申请实施例提供一种图像处理模型训练方法，该方法包括：获取训练样本集；训练样本集中的样本包括成对的低质量图像和高质量图像、以及低质量图像对应的边缘图像；质量的评价参数包括以下之一：分辨率、清晰度、色彩饱和度；利用训练样本集本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像处理模型训练方法，其特征在于，所述方法包括：获取训练样本集；所述训练样本集中的样本包括成对的低质量图像和高质量图像、以及所述低质量图像对应的边缘图像；所述质量的评价参数包括以下之一：分辨率、清晰度、色彩饱和度；利用所述训练样本集中的样本对初始网络模型进行训练，得到图像处理模型；其中，所述训练样本集中的所述边缘图像作为先验信息参与训练；所述图像处理模型包括图像超分辨率模型或图像增强模型。2.根据权利要求1所述的方法，其特征在于，获取训练样本集的步骤，包括：获取初始训练样本集，所述初始训练样本集中的样本包括：成对的低质量图像和高质量图像；针对每个样本中的低质量图像进行边缘提取，得到每个低质量图像对应的边缘图像；由每个成对的低质量图像和高质量图像，及所述低质量图像对应的边缘图像组成一个新样本，得到包含多个新样本的训练样本集。3.根据权利要求1所述的方法，其特征在于，所述初始网络模型包括以下之一：CNN网络模型、GAN网络模型或RNN网络模型。4.根据权利要求3所述的方法，其特征在于，利用所述训练样本集中的样本对所述初始网络模型进行训练，得到图像处理模型的步骤，包括：确定所述训练样本集中的每轮训练对应的训练样本子集，对于每轮训练均执行以下操作：将当前训练样本子集中每个样本中的低质量图像和边缘图像输入所述初始网络模型，得到每个所述样本对应的输出图像；根据当前训练样本子集中每个样本的输出图像与所述样本中的高质量图像计算本轮训练对应的损失函数值；根据本轮训练的损失函数值对所述初始网络模型进行反向传播训练，直到训练轮次达到预设次数或者损失函数值收敛至预设收敛阈值时停止训练，得到图像处理模型。5.根据权利要求4所述的方法，其特征在于，将当前训练样本子集中每个样本中的低质量图像和边缘图像输入所述初始网络模型，得到每个所述样本对应的输出图像的步骤，包括：针对当前训练样本子集中的每个样本，均执行以下步骤：将所述样本中的低质量图像和边缘图像分别输入所述初始网络模型；利用所述初始网络模型提取所述低质量图像对应的第一特征图和所述边缘图像对应的第二特征图，并将所述第一特征图和所述第二特征图进行合并；通过所述初始网络模型根据合并后的特征图进行特征映...

【专利技术属性】
技术研发人员：贺沁雯，
申请(专利权)人：北京金山云网络技术有限公司，
类型：发明
国别省市：