SIGAI特约作者 谢恩泽同济计算机研三在读,face 研究实习生 主要研究方向为目标检测,语义分割等其中包括文字检测和识别
前言:这篇文章是第一个做弯曲文本的端到端检测 识别。
传统的方法将文字检测和文字识别分为两个分开的部分,即输入一张图,先进行文字检测,检测出文字的位置,再进行文字识别,即对检测出的文字抠出来并送入识别网络。这样一方面比较费时间,第二没有共享检测和识别的特征。
一 什么是端到端检测识别?
即只需要一个网络,输入一张图片,同时输出检测和识别的结果,相比传统先检测再识别的好处在于
(1)速度更快,因为流程更简单
(2)效果更好,由于多任务学习使得网络能提取更为鲁棒的特征,每一个任务都可以有一定程度的提高。
端到端训练文字检测和识别可以使得这两个任务都能得到提高,使得梯度能从这两个分支分别回传到主干网络,能使得定位更加精准并减少错误样本的检测。
传统做法大多数分为以下四步
(1)检测出旋转的文字框
(2)并做一个仿射变换
(3)在feature map上将文字区域抠出来
(4)用crnn或者类似sequence的方法识别。
这种方法的缺点在于文字可能检测的不够精准,这样对于识别来说就会造成一定困难,比如文字边缘多框了一些空白区域等。
二 弯曲文本的检测和识别
弯曲文本检测:过去的方法主要解决水平文字检测或者倾斜的文字检测,而弯曲文字在自然场景中大量出现,如星巴克标志。
水平文字检测只需要检测出文字的左上角和右下角,即4个变量[x1,y1,x2,y2]
倾斜文字一般有两种方法,1是用4个点表示,即8个变量,[x1,y1,x2,y2,x3,y3,x4,y4],2是用水平矩形 旋转角度表示,需要5个变量,[x1,y1,x2,y2,theta]
而弯曲文字往往需要用更多的点才能精准描述。如下图所示。
因此,过去的方法无法很好的适应弯曲文字检测的任务。
传统文字识别的方法是把文字区域当做一个序列,用lstm从左到右扫描文字,得到文字识别结果,代表方法为CRNN,这种方法对于水平的文字效果很好,但是对于弯曲的文字效果却很差。
弯曲文本识别的难度在于水平检测框或者四边形检测框 做仿射变换,如下图所示
(1)无法精准定位文字区域,水平检测框和四边形检测框中文字区域都只占据很小的一部分,大部分都是背景,而基于分割的方法可以准确的包围弯曲文字区域。
(2)水平或者倾斜检测框无法扭正文本,因此基于lstm的CRNN识别方法效果就会很差,而用实例分割的方法可以精准检测文字并识别
三 网络结构介绍
该方法是基于Mask R-CNN的,Mask R-CNN是Facebook提出的通用物体检测分割框架,可以同时对80类物体做检测和分割,其中有三个分支
(1)classification分支,用来对目标分类
(2)bounding box regression分支,用来定位物体
(3)mask分支,用来对框内的物体做实例分割。如下图所示
这篇文章是基于mask rcnn的,流程图如下。
创新点主要是对mask分支进行了改进,其中把mask 分支做了类似于人体关键点的检测,把一个个字符当做整个文字的关键点,即不仅对bounding box内的整个文字区域做instance segmentation,同时对每个字符做分割,这里要注意的是每个字符的标注并不来源于真实数据,而是来自于人造数据集synth800k数据,因为真实数据集中往往只有单词级别的标注,并没有字符级别的标注。
这篇文章的创新之处主要在于在mask 分支加入了文字识别的分支,详细分析一下mask 分支
可以看到mask 分支一共分三部分,第一部分是global word map,第二部分是每个字符的character map,这关系到如何识别,第三部分是字符的关键点,即background map,用来区分背景和字符
(1) global word map 和标准的mask rcnn分割的分支一样,目的是分割出word的区域。
(2) character maps 该map的channel数量和要识别的文字类别一致,一般数据集为36(10数字 26字母),即对所有字符的中心做语义分割,这个字符属于哪一类就会在channel对应层上响应最高。
(3) bachground map 是定位字符的,和(2)类似,但是不需要区分是哪个字符,只需要区分是不是字符。
识别是通过一种 叫pixel-voting的方法,即通过bachground map去算他在36个字符的fmap上的响应,通过投影的方式,取响应最大的当做类别。
可以说这是一种基于语义分割的方法的文字识别。
pixel-voting做文字识别
上图 描述了(2)(3)如何做文字识别的细节。通过background map中的字符位置投影到character map上,去计算在每一层的响应值,哪一层响应最大,就认为这个字属于哪一类。主要是通过pixel-voting的方法。
通过这种方法识别的好处在于:
(1)借鉴了语义分割的思路,直接分割出字符的中心区域并判断类别
(2)抛弃了基于lstm的序列识别的方法,因此对于无论是水平文字还是弯曲文字识别效果都很鲁棒。
标签生成
训练神经网络需要label,如何准备label呢?
可以看到,左图中蓝色的水平box是rpn 产生的候选框,红色的多边形框是word的真实框,黄色的box是每个字符的真实 box框,绿色的水平框是红色多边形的最小外接矩形框。 右边:上图是mask rcnn做实例分割的label,下边是根据黄色框进行等比例缩圈得到的不同字符的语义分割的label(不同颜色代表不同字符。)
细节:其中大量的字符标注(黄色框)不来源于真实数据集,而是来自人造数据集Synth800k,其中包括了80万张人工合成图。
实验结果
可视化结果:作者在icdar2013,2015和total-text上做了实验,效果还是很不错的,红色框表示fast rcnn分支出的水平box,白点是每个字符的定位。
作者对比Textboxes CRNN这种水平box检测识别的效果,这种方法在弯曲文本上优越性的确更强。
可以看到Textboxes检测只能检测水平框,精准度很低,其次这样检测出的结果送给识别网络效果很差,很多地方都识别错了。
实验数据:
作者分别在ICDAR2013,2015和Total-Text这三个数据集上做了实验。
ICDAR2013是水平文本,ICDAR2015是多方向文本,Total-Text是弯曲文本,因此这三个数据集代表了不同类型的文字检测识别任务。
可以看到所提出的方法在弯曲文字上大幅度超过现有方法,在传统的水平和多方向文字检测识别上效果也领先现有的方法。
总结:
这个方法是一个可以端到端检测和识别弯曲文字的方法,基于目前最好的实例分割模型Mask R-CNN,对于曲形文字的端到端识别基于top-down自顶向下的方法基本是做的很好了。
此外该方法有一个缺点,对于中文识别估计效果不好,因为中文类别数太多,他们识别的feature map层数和类别成正比,因此feature map层数会很大,会很拖累速度。
不过瑕不掩瑜,该方法是第一个尝试做端到端的弯曲文字检测和识别,创新度非常高,其中在Mask分支通过关键点的思路做文字识别更是首创,值得大家多学习一下。