ocr机器学习

Multi-Oriented Scene Text Detect

2018-10-10  本文已影响0人  ce0b74704937

Multi-Oriented Scene Text Detection via Corner Localization and Region Segmentation发表于2018年的cvpr,该文章通过结合角点检测和图像分割来对图像文字进行定位

Introduction

目前文字定位方法分为两个分支,一是基于物体检测的方法(SSD,YOLO,DenseBox),如TextBoxes,FCRN,EAST等。还一种是基于图像分割的方法。本文是将上述两种方法结合,提出的一种新的检测方法。

通过文章的结果可以看出F-measure会比EAST高一点,但是速度比EAST慢。

Network

文章中的网络结构采用FPN/DSSD的网络结构,如下图所示


1.png

1. Feature Extraction
如上图红色框所示,文中采用VGG16,将fc6和fc7层换成卷积层conv6和conv7,并在后面添加了一些卷积层(conv8, conv9, conv10, conv11),用于增加感受野的范围。之后采用DSSD的top-down pathway结构,解卷积采用了从conv11到conv3的feature map(其中conv10到conv3的featrue map被重用),输出的feature命名为F_{3},F_{4},F_{7},F_{8},F_{9},F_{10},F_{11}。最后得到的conv11,和所有解卷积的feature maps用于角点和位置敏感图(position-sensitive)的预测

2. Corner Detection
对于一个旋转矩形来讲,可以通过顺时针分布的四个角点\{(x_{1},y_{1}),(x_{2},y_{2}),(x_{3},y_{3}),(x_{4},y_{4})\}来得到,这四个点位置分别为左上,右上,左下,右下。为了更方便的检测角点,文中对角点进行的新的定义,C=(x_{c},y_{c},ss,ss),其中x_{c},y_{c}代表一个角点如(x_{1},y_{1}这个左上角点),同时它也是一个水平矩形的中心点,ss代表的是待检测旋转矩形的短边。这里意思就是将待检测角点用一个水平矩形表示,角点的位置就是这个水平矩形的中心,检测出水平矩形就相当于检测出角点的位置。

通过上面角点的重新定义,检测角点的方法就可以类似于SSD和DSSD,利用定义的default boxes(类似于Faster RCNN中的anchor boxes)来进行矩形的检测。与物体检测有所不同的是,同一个位置可能存在多个角点(例如同一个位置可能同时为左下角点和右上角点)。所以对于大小为m\times n的feature map同时有k个default boxes的情况,score branch输出的类别分数(是否存在角点)通道数为k\times q \times 2,offset branch输出的通道数为k\times q \times 4,其中q代表角点的类型,默认为4。这部分如上图中的黄色框所示。

default boxes的scales设置如下表所示,其中default boxes的长宽高比为1。


表1

3. Position-Sensitive Segmentation
这部分如上图中的蓝色框所示
对于一个文本框,可以将框等分为g\times g部分,本文中分为2\times 2也就是四个区域,这部分的预测用于对上面检测出的框打分使用。下面会说明。
这部分的输出是重用了F_{3},F_{4},F_{7},F_{8},F_{9},上采用至F_{3}大小然后相加,最后连续使用两个Conv1x1- BN-ReLU-Deconv2x2 块,得到g\times g通道大小与输入图像一样的feature map。

Training and Inference

1.Training-Label Generation
对于每个输入训练样本,首先将标注转换成包围字符区域最小的矩形,然后确定4个角点的位置。

对于一个旋转矩形,4个角点的确定遵循下面两个规则:

  1. 左上与左下两个点的x坐标小于右上和右下两个点的x坐标
  2. 左上与右上两个点的y坐标小于右上和右下两个点的y坐标

通过确定好的4个角点就可以确定旋转矩形的位置了,计算旋转矩形的短边就可以使用一个水平正方形重新定义角点了,通过一件确定的旋转矩形可以很方便的求的position-sensitive segmentation掩码,所得的label结果如下图所示

图2.png

2.Training-Optimization
损失函数如下:
L= \frac{1}{N}L_{conf}+\frac{\lambda_{1}}{N_{c}}L_{loc}+\frac{\lambda_{2}}{N_{s}}L_{seg}

其中,L_{conf}L_{loc}表示角点检测中score branch和offset branch输出的loss,L_{seg}表示position-sensitive segmentation的损失函数。N_{c}表示正例default boxes的个数,N_{s}表示分割maps中的像素个数(分割)。\lambda_{1}\lambda_{2}为loss函数的平衡因子,文中分别取值为1和10。

L_{conf}采用的是交叉熵计算
L_{conf}=CrossEntropy(y_{c},p_{c})
y_{c}表示真实标签,p_{c}表示预测值

L_{loc}采用smooth L1 loss计算:
L_{loc}=SmoothL1(y_{l},p_{l})
y_{l}p_{l}定义同Fast RCNN

L_{seg}采用Dice loss:
L_{seg}=1-\frac{2y_{s}p_{s}}{y_{s}+p_{s}}
y_{s}表示label值,y_{s}表示预测值

3.Inference-Sampling and Grouping
在通过角点检测得到一系列的角点后,将score小于0.5的角点忽略,然后使用NMS得到最终的候选角点信息。
因为一个旋转矩形能通过两个点和垂直与两点连线的一条边确定,在预测各个角点时短边长度已知,这时将所有候选角点与对应短边组合就可以得到一系列的候选矩形了。
在得到候选矩形的过程中有一些过滤的规则:

  1. 比如左上角点的x坐标是必须小于右上角点的x坐标
  2. 旋转矩形的短边必须大于某个阈值,文中设为5
  3. 两个待组合点预测的短边ss_{1},ss_{2}必须满足\frac{max(ss_{1},ss_{2})}{min(ss_{1},ss_{2})}\le 1.5

4. Inference-Scoring
对于得到的候选矩形,将其划分为g\times g,然后利用segmentation maps,将属于每一块(总共g\times g块)的segmentation对应的像素相加后求平均,最后将g\times g个平均值再求平均值得到候选矩形框的分数,此过程如下图所示。
设定阈值为0.6,旋转矩形分数小于阈值的被过滤

图3.png

到这基本就是该文章的所有原理了。

中文本定位与识别的评测方法

欢迎加入OCR交流群:785515057(此群已满)
欢迎加入OCR交流群2:826714963

上一篇下一篇

猜你喜欢

热点阅读