【技术实现步骤摘要】
图像修复方法、装置和电子设备
本公开涉及文本生成领域,尤其涉及一种图像修复方法、装置、电子设备及计算机可读存储介质。
技术介绍
文字作为人类专利技术的抽象的交流符号,具有丰富的表达性,并在自然场景中作为信息表达大量出现。由于文字含有丰富的语义信息,识别自然场景中的文字成为大量视觉方面应用的基础,如目标定位、人机交互、图像搜索、机器导航和工业自动化等等。因此,自然场景下对文字的识别和理解是近年的研究与应用的热点之一。现有技术中存在需要去除图像中的噪声的需求,如有些图像上有一些字幕,后续需要使用无字幕的图像时,需要将字幕从图像上去除掉并将字幕部分补充成图像的颜色。现有技术中的去除图像中的文字的方法,一般是先使用阈值法提取图像中的文字,之后用图像减法减去文字部分的颜色得到去除文字之后的图像;或者是先用阈值法图区图像中的文字,之后使用OPENCV中inpaint()函数对图像修复。但是上述文字去除的方法得到去除结果都不够精细,去除掉为之后原先文字区域的颜色不自然。
技术实现思路
提供该
技术实现思路
...
【技术保护点】
1.一种图像修复方法,其特征在于,包括:/n获取待修复图像;/n生成待修复图像中的噪声区域的修复图像;/n将所述噪声区域的修复图像贴合到所述待修复图像上得到修复后的图像。/n
【技术特征摘要】
1.一种图像修复方法,其特征在于,包括:
获取待修复图像;
生成待修复图像中的噪声区域的修复图像;
将所述噪声区域的修复图像贴合到所述待修复图像上得到修复后的图像。
2.如权利要求1所述的图像修复方法,其中所述生成待修复图像中的噪声区域的修复图像,包括:
根据所述待修复图像生成噪声区域的轮廓;
根据所述待修复图像生成中间图像;
根据所述噪声区域的轮廓以及中间图像生成所述噪声区域的修复图像。
3.如权利要求2所述的图像修复方法,其中所述生成待修复图像中的噪声区域的修复图像,包括:将所述待修复图像输入图像修复模型得到所述噪声区域的修复图像,其中所述修复模型包括编码层、第一解码层和第二解码层,其中:
所述将所述待修复图像输入所述编码层得到所述待修复图像的特征图;
其中,所述根据所述待修复图像生成噪声区域的轮廓,包括:
将所述待修复图像的特征图输入所述第一解码层得到所述噪声区域的轮廓;
所述根据所述待修复图像生成中间图像,包括:
将所述待修复图像的特征图输入所述第二解码层得到所述中间图像。
4.如权利要求3所述的图像修复方法,其中所述图像修复模型通过以下训练步骤得到:
获取训练数据集合,其中所述训练数据集合中的训练数据是由原始图像、在原始图像中添加了噪声的第一图像以及由所述噪声生成的第二图像所组成的三元数据;
根据所述原始图像以及所述第二图像生成噪声区域的修复图像;
初始化图像修复模型的参数;
将所述第一图像输入所述图像修复模型得到预测的噪声区域的修复图像;
根据所述噪声区域的修复图像以及所述预测的噪声区域的修复图像计算误差;
基于所述误差更新所述图像修复模型的参数;
使用更新后的参数迭代上述参数更新的过程直至达到收敛条件;
将达到收敛条件时所得到的参数作为训练好的图像修复模型的参数。
5.如权利要求4所述的图像修复方法,其中所述获取训练数据集合,包括:
获取原始图像;
在原始图像上随机生成噪声得到第一图像;
根据所述噪声生成仅包含所述噪声的第二图像。
6.如权利要求5所述的图像修复方法,其中所述第一图像中的噪声的像素点与所述噪声图像中的噪声像素点一一对应。
7.如权利要求1-6任一项所述的图像修复方法,其中所述第二图像为背景为黑色、噪声为白色的二值化图像。
8.如权利...
【专利技术属性】
技术研发人员:周恺卉,王长虎,
申请(专利权)人:北京字节跳动网络技术有限公司,
类型:发明
国别省市:北京;11
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。