[发明专利]一种图像生成方法及装置

说明书

本发明公开一种图像生成方法及装置，所述方法包括：获取待处理视频以及目标服饰图像；从所述待处理视频中提取连续的至少两个视频帧图像；所述视频帧图像包括目标对象；对所述至少两个视频帧图像进行目标对象分割处理，得到所述目标对象的二值图；融合所述目标服饰图像、二值图以及所述至少两个视频帧图像中的目标视频帧图像，得到目标融合图像。本发明实现了在背景复杂的情况下依然可以使目标对象与指定服饰图片精准的匹配，输出更加自然的证件照。

技术领域

本发明涉及计算机图像处理技术领域，特别涉及一种图像生成方法及装置。

背景技术

目前人们获取自身穿正装的证件照，不仅要花钱购买或租赁正装，还需要到专业的证件照拍摄场所进行拍摄，耗时耗财。随着深度学习技术的快速发展，基于深度学习的证件照生成方法能够自动生成证件照，省时省力。

现有技术中，在单帧图像上利用传统的图像分割方法和抠图进行证件照合成，此方法通过固定的数学运算区分前后背景，容易受背景的复杂度影响，在一些非纯色背景下，前景和背景容易出现交叉错分，很大程度上影响最终的合成效果，而且通过抠图其最终的合成效果仅为背景替换，而非服装更换，如果证件照要求穿正装，则依然需要穿着正装。

此外，当遇到背景比较复杂，如头发周围颜色与发色相近、图像曝光、模糊等，分割出来的人体区域与背景区域混合，效果不好，此时存在分割出来的人体区域与服饰图片融合效果差等问题。

发明内容

为了解决现有技术的问题，本发明实施例提供了一种图像生成方法及装置。所述技术方案如下：

一方面，提供了一种图像生成方法，所述方法包括：

获取待处理视频以及目标服饰图像；

从所述待处理视频中提取连续的至少两个视频帧图像；所述视频帧图像包括目标对象；

对所述至少两个视频帧图像进行目标对象分割处理，得到所述目标对象的二值图；

融合所述目标服饰图像、二值图以及所述至少两个视频帧图像中的目标视频帧图像，得到目标融合图像。

另一方面，提供了一种图像生成装置，所述装置包括：

获取模块，用于获取待处理视频以及目标服饰图像；

图像提取模块：用于从所述待处理视频中提取连续的至少两个视频帧图像；所述视频帧图像包括目标对象；

图像分割模块：用于对所述至少两个视频帧图像进行目标对象分割处理，得到所述目标对象的二值图；

图像融合模块：用于融合所述目标服饰图像、二值图以及所述至少两个视频帧图像中的目标视频帧图像，得到目标融合图像。

进一步地，所述图像分割模块包括：

目标对象分割模块：用于将所述至少两个视频帧图像输入图像分割网络，通过所述图像分割网络对所述至少两个视频帧图像进行目标对象分割处理，得到所述目标对象的二值图。

进一步地，所述装置还包括：

第一训练数据获取模块：用于获取第一训练数据；所述第一训练数据包括第一图像组和所述第一图像组对应的参考二值图，所述第一图像组中包括连续的至少两个样本视频帧图像；

第一网络模型输入模块：用于将所述第一图像组输入至第一初始神经网络模型中进行图像分割训练，得到输出的训练二值图；

第一网络模型损失值确定模块：用于根据所述训练二值图与所述参考二值图的二阶范数确定第一损失值；

第一网络模型训练模块：用于按照最小化所述第一损失值的方向调整所述第一初始神经网络模型中的模型参数直至满足训练结束条件，得到所述图像分割网络。