图像处理方法、图像处理模型的训练方法和装置制造方法及图纸

技术编号:28132028 阅读:28 留言:0更新日期:2021-04-19 11:55
本发明专利技术提供了一种图像处理方法、图像处理模型的训练方法和装置,基于待处理图像中水印所在的位置,生成指示待处理图像中包含水印的图像区域的蒙版;将待处理图像和蒙版输入至图像处理模型中,输出去除水印后的图像,该图像处理模型基于样本集训练得到,该样本集包括样本图像和样本图像的样本蒙版,该样本集中的一部分样本图像中样本蒙版指示的图像区域以外的区域中包含有文字。该方式通过图像处理模型去除待处理图像中蒙版对应的图像区域的水印,由于训练图像处理模型的样本集中存在样本蒙版指示的图像区域以外的区域中包含文字的样本图像,使得图像处理模型具有处理文字的能力,在补全水印区域的图像时避免生成伪影,从而提高了水印的去除效果。而提高了水印的去除效果。而提高了水印的去除效果。

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


[0001]本专利技术涉及图像处理
,尤其是涉及一种图像处理方法、图像处理模型的训练方法和装置。

技术介绍

[0002]各种平台的创作者或发布者在发布图像时通常会加上水印,水印的存在既影响了图像整体的美感和氛围,又影响了其他用户对图像的使用。因而,在用户使用带有水印的图像时,需要去除该图像中的水印,通常在去除图像上的水印时,首先检测待处理图像中的水印位置,然后通过图像补全方法生成该水印位置对应的补全内容,并使用生成的补全内容替换待处理图像中的水印位置对应的内容,得到水印去除后的图像。
[0003]相关技术中,图像补全方法包括传统图像补全方法和基于深度学习的图像补全方法,传统图像补全方法通常基于图像块的相似性,依据待补全位置周围已知像素推断缺失像素,该方式难以生成复杂纹理,对于图像复杂时的补全效果较差。基于深度学习的图像补全方法从训练集的样本图像中学习到图像补全的能力,但该方式在待处理图像中包含有文字或者存在遗留水印像素时,在补全图像过程中容易受文字像素或水印像素的影响生成伪影,导致图像补本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种图像处理方法,其特征在于,所述方法包括:基于待处理图像中水印所在的位置,生成所述待处理图像的蒙版;其中,所述蒙版用于指示所述待处理图像中包含所述水印的图像区域;将所述待处理图像和所述蒙版输入至预先训练完成的图像处理模型中,输出去除所述水印后的所述待处理图像;其中,所述图像处理模型基于预设的样本集训练得到,所述样本集包括样本图像和所述样本图像对应的样本蒙版,所述样本集中的一部分样本图像中,所述样本蒙版指示的图像区域以外的区域中包含有文字。2.根据权利要求1所述的方法,其特征在于,所述一部分样本图像通过下述方式获得:从所述样本集中抽取样本图像,将抽取的样本图像作为目标图像;在所述目标图像中加入文字,得到文字图像;基于所述文字图像和所述目标图像对应的样本蒙版,生成所述样本蒙版指示的图像区域以外的区域中包含有文字的样本图像。3.根据权利要求2所述的方法,其特征在于,所述基于所述文字图像和所述目标图像对应的样本蒙版,生成所述样本蒙版指示的图像区域以外的区域中包含有文字的样本图像的步骤,包括:从所述文字图像中确定所述目标图像对应的样本蒙版的图像区域;将所述文字图像中确定出的所述图像区域替换为所述目标图像对应的样本蒙版的图像区域,将替换后的所述文字图像作为所述样本蒙版指示的图像区域以外的区域中包含有文字的样本图像。4.根据权利要求1所述的方法,其特征在于,所述将所述待处理图像和所述蒙版输入至预先训练完成的图像处理模型中,输出去除所述水印后的所述待处理图像的步骤,包括:将所述待处理图像和所述蒙版输入至所述图像处理模型;通过所述图像处理模型,在所述待处理图像中去除所述蒙版对应的图像区域,得到缺失图像,并对所述缺失图像进行图像补全,得到去除所述水印后的所述待处理图像。5.一种图像处理模型的训练方法,其特征在于,所述训练方法包括:获取样本集;所述样本集包括样本图像和所述样本图像对应的样本蒙版,所述样本集中的一部分样本图像中,所述样本蒙版指示的图像区域以外的区域中包含有文字;基于所述样本集对初始模型进行训练,得到图像处理模型。6.根据权利要求5所述的训练方法,其特...

【专利技术属性】
技术研发人员:贺沁雯李果鲁方波汪贤樊鸿飞蔡媛
申请(专利权)人:北京金山云网络技术有限公司
类型:发明
国别省市:

网友询问留言 已有0条评论
  • 还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。

1