【技术实现步骤摘要】
抠图方法、抠图装置和电子设备
[0001]本申请涉及图像处理
,尤其涉及一种抠图方法、抠图装置和电子设备。
技术介绍
[0002]抠图技术(Image Matting)是将图像中的前景从背景中分离出来的技术,该技术能够将人们需要的区域提取出来,便于对图像关键信息进行后续操作。
[0003]现有技术通常采用三分草图(Trimap)技术进行的高性能数字抠图,然而,该技术需人工绘制或预测三分草图或原始图像的原始背景作为额外输入,不但浪费人力和算力,还可能存在与实际应用场景不符的问题,阿降低用户抠图体验。
技术实现思路
[0004]本申请实施例提供了一种抠图方法、抠图装置和电子设备,以改善上述问题。
[0005]第一方面,提供一种抠图方法,包括:获取原始图像的前景分割遮罩图和深度图;将原始图像和所述分割遮罩图输入第一前景预测网络中,得到第一前景预测图,将原始图像和所述深度图输入第二前景预测网络中,得到第二前景预测图;将原始图像与第一前景预测图输入到互补学习网络中,得到第一误差图,将原始图像与第二 ...
【技术保护点】
【技术特征摘要】
1.一种抠图方法,其特征在于,包括:获取原始图像的前景分割遮罩图和深度图;将所述原始图像和所述分割遮罩图输入第一前景预测网络中,得到第一前景预测图,将所述原始图像和所述深度图输入第二前景预测网络中,得到第二前景预测图;将所述原始图像与所述第一前景预测图输入到互补学习网络中,得到第一误差图,将所述原始图像与所述第二前景预测图输入到所述互补学习网络中,得到第二误差图;比较所述第一误差图与所述第二误差图在同一图像分区的置信度,根据比较结果,对所述第一前景预测图或所述第二前景预测图位于该图像分区的图像进行修正,获得修正后的第一前景预测图和修正后的第二前景预测图。2.根据权利要求1所述的方法,其特征在于,所述将所述原始图像和所述分割遮罩图输入到第一前景预测网络中,得到第一前景预测图,包括:利用所述原始图像作为标签,采用生成对抗方式对所述第一前景预测网络进行训练,得到所述第一前景预测图;所述将所述原始图片和所述深度图输入到第二前景预测网络中,得到第二前景预测图,包括:利用所述原始图像作为标签,采用生成对抗方式对所述第二前景预测网络进行训练,得到所述第二前景预测图。3.根据权利要求2所述的方法,其特征在于,所述利用所述原始图像作为标签,采用生成对抗方式对所述第一前景预测网络进行训练,得到所述第一前景预测图,包括:利用所述原始图像的前景真值监督所述第一前景预测网络,得到第一前景预测图;生成器将所述第一前景预测图与其他背景图合成第一更新图片;利用判别器判别所述第一更新图片的真实可信性,以监督所述第一前景预测网络输出新的第一前景预测图。4.根据权利要求3所述的方法,其特征在于,所述利用所述原始图像作为标签,采用生成对抗方式对所述第二前景预测网络进行训练,得到所述第二前景预测图,包括:利用所述原始图像的前景真值监督所述第二前景预测网络,得到第二前景预测图;生成器将所述第二前景预测图与其他背景图合成第二更新图片;利用判别器判别所述第二更新图片的真实可信性,以监督所述第二前景预测网络输出新的第二前景预测图。5.根据权利要求4所述的方法,其特征在于,将所述原始图像与所述第一前景预测图输入到互补学习网络中,得到第一误差图,将所述原始图像与所述第二前景预测图输入到所述互补学习网络中,得到第二误差图,包括:将所述原始图像、所述第一前景预测图、以及所述第二前景预测图输入到互补学习网络中,对所述互补学习网络进行有标签的预训练;继续向训练完成的所述互补学习网络中输入更新后的第一前景预测图,并根据所述前景真值得到所述第一误差图;继续向训练完成的所述互补学习网络中输入更新后的第二前景预测图,并根据所述前景真值得到所述第二误差图。6.根据权利要求5所述的方法,其特征在于,所述第一误差图和所述第二误差图均为灰...
【专利技术属性】
技术研发人员:徐博,黄晗,郭彦东,
申请(专利权)人:OPPO广东移动通信有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。