【技术实现步骤摘要】
一种图像风格迁移方法、装置、电子设备及存储介质
本公开涉及图像数据处理
，尤其涉及一种图像风格迁移方法、装置、电子设备及存储介质。
技术介绍
目前，用户可以使用手机等终端设备进行拍照或视频拍摄等，然后可以根据不同的喜好将图像(视频可以视为连续图像的集合)转换为不同风格效果，进而实现用户不同的风格体验。一般的，终端设备视频进行不同风格的迁移转换，其风格模型的计算量通常与视频帧图像的分辨率直接相关。为减少终端计算量、提高终端处理效率，目前在人像图像风格迁移处理时，通常会对人脸区域进行裁剪后单独处理，然后再融合到原图像中。然而，这样常常会造成图像中局部区域风格一致性较差，出现过渡不自然引起的视觉效果较差的问题。
技术实现思路
本公开提供一种图像风格迁移方法、装置、电子设备及存储介质，以至少解决相关技术中图像风格迁移后局部图像风格一致性较差的技术问题。本公开的技术方案如下：根据本公开实施例的第一方面，提供一种图像风格迁移方法，包括：将待处理图像中的人脸区域图像输入人脸风格迁移模型，得...
【技术保护点】
1.一种图像风格迁移方法，其特征在于，包括：/n将待处理图像中的人脸区域图像输入人脸风格迁移模型，得到人脸风格化图像；/n获取所述待处理图像的全图下采样图像，将所述全图下采样图像输入全图风格迁移模型，得到与所述人脸风格化图像的风格类型相同的全图风格化图像；/n将所述全图风格化图像上采样至所述待处理图像的大小，得到全图上采样图像；/n根据所述全图上采样图像对应的第一灰度图，对所述待处理图像对应的第二灰度图中像素的灰度值进行调整，得到亮度迁移图像；/n将所述亮度迁移图像与所述人脸风格化图像进行融合，得到目标风格化图像。/n
【技术特征摘要】
1.一种图像风格迁移方法，其特征在于，包括：
将待处理图像中的人脸区域图像输入人脸风格迁移模型，得到人脸风格化图像；
获取所述待处理图像的全图下采样图像，将所述全图下采样图像输入全图风格迁移模型，得到与所述人脸风格化图像的风格类型相同的全图风格化图像；
将所述全图风格化图像上采样至所述待处理图像的大小，得到全图上采样图像；
根据所述全图上采样图像对应的第一灰度图，对所述待处理图像对应的第二灰度图中像素的灰度值进行调整，得到亮度迁移图像；
将所述亮度迁移图像与所述人脸风格化图像进行融合，得到目标风格化图像。
2.根据权利要求1所述的方法，其特征在于，所述根据所述全图上采样图像对应的第一灰度图，对所述待处理图像对应的第二灰度图中像素的灰度值进行调整，得到亮度迁移图像包括：
分别计算所述第一灰度图和第二灰度图中像素值的均值和/或方差；
基于所述均值和/或方差计算所述第二灰度图中像素的目标像素值；
将所述第二灰度图中像素的像素值更新为所述目标像素值，得到亮度迁移图像。
3.根据权利要求1所述的方法，其特征在于，所述根据所述全图上采样图像对应的第一灰度图对，所述待处理图像对应的第二灰度图中像素的灰度值进行调整，得到亮度迁移图像包括：
将所述全图上采样图像转换为第一YUV图像，提取所述第一YUV图像的Y通道，得到第一灰度图，以及将所述待处理图像转换为第二YUV图像，提取所述第二YUV图像的Y通道，得到第二灰度图；
根据所述第一灰度图对所述第二灰度图中像素的灰度值进行调整，得到调整后Y通道；
将所述调整后Y通道与所述第二YUV图像的UV通道合并，得到调整后YUV图像；
将所述调整后YUV图像转换为与所述待处理图像相同的图像格式，得到亮度迁移图像。
4.根据权利要求2所述的方法，其特征在于，当所述分别计算所述第一灰度图和第二灰度图中像素值的均值和/或方差为分别计算得到所述第一灰度图和第二灰度图中像素值的均值和方差时，所述基于所述均值和/或方差计算所述第二灰度图中像素的目标像素值包括：
上式中，Yij为所述第二灰度图中第i行第j列像素的像素值，Y′ij为与所述Yij对应的目标像素值，mean2为所述第二灰度图中像素值的均值，var2为所述第二灰度图中像素值的方差，mean1为所...
【专利技术属性】
技术研发人员：张雷，方慕园，
申请(专利权)人：北京达佳互联信息技术有限公司，
类型：发明
国别省市：北京;11
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。