买专利,只认龙图腾
首页 专利交易 科技果 科技人才 科技服务 商标交易 会员权益 IP管家助手 需求市场 关于龙图腾
 /  免费注册
到顶部 到底部
清空 搜索

【发明授权】一种基于上下边界提取的自然场景下文字检测方法_南京大学_202110265373.1 

申请/专利权人:南京大学

申请日:2021-03-11

公开(公告)日:2024-04-19

公开(公告)号:CN112861860B

主分类号:G06V10/25

分类号:G06V10/25;G06V10/44;G06V30/19;G06N3/0464;G06N3/08;G06T7/11

优先权:

专利状态码:有效-授权

法律状态:2024.04.19#授权;2021.06.15#实质审查的生效;2021.05.28#公开

摘要:本发明公开了一种基于上下边界提取的自然场景下文字检测方法,涉及计算机视觉文字检测和图像分割领域。具体包括:步骤1,通过卷积神经网络模型提取图像的特征图,并通过多尺度特征金字塔模型对特征图进行增强;步骤2,通过语义分割模型预测文字的中心区域,上边界区域,下边界区域和文字的整体区域;步骤3,通过广度优先搜索将中心区域逐步扩展至整体区域,同时通过对上下边界区域进行滤波操作得到文字的形式化上下边界线;步骤4,组合文字的中心区域、整体区域以及上下边界线得到最终的文字检测结果。优点在于能够提取图像中文字的上下边界并定位文字区域,能够进行自然场景下的图像文字检测。

主权项:1.一种基于上下边界提取的自然场景下文字检测方法,其特征在于,包括如下步骤:步骤1,通过卷积神经网络模型提取输入图像的特征图,并通过多尺度特征金字塔模型对特征图进行增强;步骤2,通过语义分割模型预测文字的整体区域、中心区域、上边界区域以及下边界区域;步骤3,通过广度优先搜索将中心区域逐步扩展至整体区域,同时通过对上下边界区域进行滤波操作得到文字的形式化上下边界线;步骤4,组合文字的中心区域、整体区域以及上下边界线得到最终的文字检测结果;其中,步骤2中,通过如下步骤建立和训练基于卷积神经网络的语义分割模型:步骤2-1,根据输入图像中的文字标注多边形框,生成其整体文字区域分割图:首先生成一张与输入图像大小一样的分割目标图,其次将文字区域填充为1,非文字区域填充为0,最终得到文字整体区域的分割目标图;步骤2-2,根据输入图像中的文字标注多边形框,生成其文字中心区域分割图,具体的,首先生成一张与输入图像大小一样的分割目标图,然后将原始的文字标注多边形框向内部收缩d个像素,收缩规则如下所示: 其中,Area为面积函数,Perimeter为周长函数,bbox为文字框,r为放缩尺度;收缩后的多边形框即为文字中心区域标注的多边形框,将收缩后的多边形框内部填充为1,外部填充为0,则得到文字中心区域的分割目标图;步骤2-3,首先计算文字标注框的上边界区域和下边界区域分别对应的上边界点topi和下边界点boti,计算如下:top_boundi=topi*1-α+boti*αbot_boundi=boti*1-α+topi*α其中,α是一个超参数,代表定义的文字上下边界区域的宽度;由此,top_boundi和topi组成了上边界相对应的一组边界点,同理得到下边界相对应的一组边界点;将计算得到的上下边界点分别连接起来,即为文字的上下边界区域,将得到的上下边界区域内部填充为1,外部填充为0,得到文字上下边界区域的分割目标图;步骤2-4,将步骤1中所得到的大小为的特征图输入语义分割模型中,经过由卷积层,上采样层和另一个卷积层构成的分割模块,最终输出大小为batchsize,4,H,W的分割预测结果图;其中,4为4个分割预测结果,即文字的整体区域、中心区域、上边界区域和下边界区域;步骤2-5,建立损失函数,将步骤2-4中得到的分割预测结果pred与步骤2-1~2-3中得到的分割目标target计算loss:loss=DiceLosspred,target具体的,DiceLoss由以下步骤计算,首先计算dice系数dice,coefficient: 则DiceLoss由以下公式计算得出:DiceLosspred,target=1-DiceCoefficientpred*M,target*M其中,M表示训练过程中的梯度回传掩码,由步骤1-2计算得到;分别在四个区域上计算对应的损失函数,最后将它们加权求和:Loss=λtextlosstext+λkernellosskernel+λtoplosstop+λbotlossbot其中λ分别表示不同区域的训练权重,loss分别表示不同区域的训练损失值,计算结束即得到分割预测结果与分割目标之间的损失值;步骤2-6,建立优化函数,使用基于随机梯度下降的优化方法对得到的损失值计算梯度并相应更新卷积神经网络模型和语义分割模型中的参数;步骤3中,通过如下步骤进行测试,将分割预测结果的中心区域逐步扩展至整体区域,并通过对上下边界区域进行滤波操作得到文字的形式化上下边界线:步骤3-1,对任意自然场景图像经过卷积神经网络和多尺度特征金字塔模型提取特征图,然后经过上采样操作得到大小为batchsize,4,H,W的分割预测结果图;步骤3-2,定位文字中心区域,设定一个固定阈值σ=0.8对中心区域分割预测结果图进行二值化,即对中心区域分割图的每个像素进行以下计算: 其中,kernel表示中心区域分割图,将二值化后的中心区域分割图计算其连通分量,计算后产生的不同连通分量再滤除掉其中像素面积小于阈值μ的连通分量,最终保留的连通分量结果即为不同的文字实例的中心区域分割结果;步骤3-3,以步骤3-2得到的中心区域分割结果图为开始,使用广度优先搜索向外扩展至文字整体区域;得到不同文字实例的整体区域分割结果图;步骤3-4,定位文字上下边界区域,以文字上边界区域为例,将上边界区域分割预测结果图与步骤3-3中得到的不同文字实例的整体区域分割结果图进行按位相与操作:topi,j=topi,jtexti,j其中,top表示上边界区域分割图,text表示由步骤3-3得到的不同文字实例的整体区域分割结果图;计算结果即是每个文字实例的上边界分割结果,同理得到每个文字实例的下边界分割结果,然后对该上下边界分割结果进行细化;步骤4具体包括:步骤4-1,对每一个由文字中心区域经步骤3-3扩展得到的文字整体区域,通过与上下边界区域进行按位相与的操作找到该文字对应的上下边界线;步骤4-2,对步骤4-1产生的所有文字整体区域、文字中心区域、上下边界线组合成对,并根据规则过滤掉不符合条件的文字;其中,过滤规则为去除没有上边界线或者下边界线的文字;步骤4-3,保留下来的文字有各自的上边界线和下边界线以及对应的在边界线上均匀采样的k个顶点,将每个文字的2*k个顶点作为该文字的轮廓点,生成了该文字基于上下边界的形式化表示,得到该文字的检测结果;其中,步骤3-4中的细化操作包括生成一个3×1的滤波器F,其具体参数为: 经过以下运算:Boundary=top·F0得到文字上边界区域细化后的边界线,保留像素值大于0的像素为文字上边界区域细化后的边界线结果,将得到的边界线像素根据其横坐标从小到大排序并均匀地从中采样k个点,这k个点即是文字上边界区域的形式化边界线;同理求得下边界线。

全文数据:

权利要求:

百度查询: 南京大学 一种基于上下边界提取的自然场景下文字检测方法

免责声明
1、本报告根据公开、合法渠道获得相关数据和信息,力求客观、公正,但并不保证数据的最终完整性和准确性。
2、报告中的分析和结论仅反映本公司于发布本报告当日的职业理解,仅供参考使用,不能作为本公司承担任何法律责任的依据或者凭证。