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

本公开关于一种图像处理模型的训练方法，包括：将样本图像输入至目标模型进行预测，得到第一预测图像，其中，目标模型是对第一预设机器学习模型进行训练得到的；从第一预测图像中获取图像频率高于预设频率的第一目标图像，其中，图像频率用于表征图像的像素灰度值在空间中的变化情况；基于样本图像和第一目标图像对第二预设机器学习模型进行训练，得到图像处理模型，其中，图像处理模型的参数量和计算量少于目标模型的参数量和计算量，图像处理模型用于图像的预测处理。用于图像的预测处理。用于图像的预测处理。

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


[0001]本公开涉及机器学习领域，尤其涉及一种神经网络模型的压缩方法及装置、电子设备。

技术介绍

[0002]近年来，生成对抗神经网络(GAN)已经在图像生成等多个领域取得了重大突破，在图像风格化，图像编辑等任务中进行了广泛的应用并取得了良好的效果。然而，已有的GAN模型往往有着大量的参数量和计算量。因此，这类模型在计算、存储能力受限的设备上难以运行，这严重限制了GAN模型的应用落地。
[0003]图像风格化是GAN模型最常用的应用之一，它可以对于输入图像进行一些风格上的转换，例如将普通的自然图片转换成手绘风格的图片等。然而，标准的图像风格化模型(例如，循环对抗生成网络CycleGAN)对于单张图片的处理需要约56.8Billion的乘加运算，X Million的参数量。一般来说，目前常见的手机设备计算、存储能力都比较有限。因此，如果直接将标准的图像风格化模型在手机设备部署，会导致该模型的运行速度极为缓慢。
[0004]目前生成对抗网络模型由于参数量和计算量较大，在计算、存储能力受限的设本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像处理模型的训练方法，其特征在于，包括：将样本图像输入至目标模型进行图像预测，得到第一预测图像，其中，所述目标模型是对第一预设机器学习模型进行训练得到的；从所述第一预测图像中获取图像频率高于预设频率的第一目标图像，其中，所述图像频率用于表征图像的像素灰度值在空间中的变化情况；基于所述样本图像和所述第一目标图像对第二预设机器学习模型进行训练，得到图像处理模型，其中，所述图像处理模型的参数量和计算量少于所述目标模型的参数量和计算量，所述图像处理模型用于图像的预测处理。2.根据权利要求1所述的方法，其特征在于，所述第一预设机器学习模型和所述第二预设机器学习模型均包括：生成对抗网络模型。3.根据权利要求1所述的方法，其特征在于，基于所述样本图像和所述第一目标图像对第二预设机器学习模型进行训练，得到图像处理模型，包括：将所述样本图像和所述第一目标图像输入至所述第二预设机器学习模型进行图像预测，得到第二预测图像；从所述第二预测图像中获取图像频率高于所述预设频率的第二目标图像；确定所述第一目标图像和所述第二目标图像的损失函数；对所述损失函数进行优化，在所述损失函数的值低于预设阈值的情况下，得到所述图像处理模型。4.根据权利要求3所述的方法，其特征在于，从所述第一预测图像中获取图像频率高于预设频率的第一目标图像，以及从所述第二预测图像中获取图像频率高于所述预设频率的第二目标图像，包括：分别对所述第一预测图像和所述第二预测图像进行频率分解，得到不同频率的第一预测子图像和不同频率的所述第二预测子图像；分别从不同频率的所述第一预测子图像和不同频率的所述第二预测子图像中筛选出所述第一目标图像和所述第二目标图像。5.根据权利要求4所述的方法，其特征在于，分别对所述第一预测图像和所述第...

【专利技术属性】
技术研发人员：张林峰，陈昕，涂小兵，
申请(专利权)人：北京达佳互联信息技术有限公司，
类型：发明
国别省市：