图像文字识别初探(一)-CRNN(Convolution Recurrent Neural Network)和DTRN(Deep-text Recurrent Network)

最近参加了一个图像文字识别比赛,之前没有接触过这个方向,所以经过多方查找,目前选定五篇论文,希望能从这五篇论文中找出能解决的方法。

图像文字识别初探(一)-CRNN(Convolution Recurrent Neural Network)和DTRN(Deep-text Recurrent Network)

图像文字识别初探(二)-FAN(Focusing Attention Network)

图像文字识别初探(三)-Mask TextSpotterr

图像文字识别初探(四)-- single-shot text detector

CRNN(Convolution Recurrent Neural Network)

论文:An End-to-ENd Trainable Neural Network for Image-based Sequence Recognition and Its application to Scene Text Recognition (2015)

这篇论文来自于华科白翔老师团队,将特征提取(feature extraction),序列模型(sequence modeling)和transcription整合到一个完整的网络,这个神经网络则被称为CRNN(Convolution Recurrent Neural Network)尽管这个网络包含好几种网络架构,但是它可以在一个损失函数下进行训练,CRNN网络的整体架构如下图所示,从下往上看,CNN从给定的图片中提取特征序列(feature sequence),在CNN网络之上,RNN(bi-LSTM)将根据卷积层生成的特征序列做预测,最后,在CRNN之上的the transcription layer将RNN层的预测结果翻译成一个label sequence。

特征提取(Feature Sequence Extraction)

在CRNN中卷积组件是由卷积和最大池化层组成,也就是标准卷积网络中去除全连接层,卷积层的作用是从输入图像中提取特征序列特征序列(feature sequence)中的每一个特征向量(feature vector)是从特征图(feature maps)中从左向右按列生成,也就是说,第i-th个特征向量是由所有特征图第i列生成的特征向量连接而成,每一列的宽度被设置为一个像素。因为卷积层,最大池化和按元素的激活函数的平移不变性(translation invariant),所以每一列的特征图对应着原图中的一个矩形区域,这个矩形区域则被称为receptive field,而且这个矩形区域是与其在特征图中相对应的从左向右的列是相同顺序的,那么就可以认为特征序列中的每一个特征向量也是与原图中的每一个receptive field相对应的,如下图所示,输入图像是要固定为同一height值的。

图像文字识别初探(一)-CRNN(Convolution Recurrent Neural Network)和DTRN(Deep-text Recurrent Network)_第1张图片

序列标注(Sequence Labeling)

在卷积层后接上一个双向RNN(bidirectional RNN)作为recurrent layer, RNN是将每一个在卷积层生成的特征序列\mathbf{x}=x_{1},\cdots, x_{T}中的x_{t}预测为一个label distribution y_{t}选择RNN的原因一是在于RNN是有很强的能力捕捉到一个序列的上下文信息,在上述的特征提取中,可以看到一个宽字符可能有好几个连续的receptive field描述,对基于图像的序列识别使用上下文比将单个字符单独对待要更加有效,而且对于一些含糊的字符,观察其上下文信息后也会很好区分;而是因为RNN也能够后向传播(back-propagates)进行权重更新,从而使得可以将CNN和RNN连接成一个完整的网络;三是因为RNN可以处理任意长度的序列,上文中对输入图片的高进行固定,是固定每一个receptive field的大小(每一个receptive field的宽为一个像素,也是固定的),那么这样的话,可以对任意宽度的图像进行处理。

CRNN中的RNN架构选择的是双向LSTM,这里不再赘述

Transcription

Transcription的作用是将RNN生成的预测转化成一个label sequence。论文里采用的CTC(Connectionist Temporal Classification)层中定义的条件概率,该概率是指在RNN生成的预测\mathbf{y}=y_{1}, \cdots, y_{T}的条件下得到label sequence \mathbf{l}的概率。因此就可以使用该概率的负log-likelihood作为目标函数进行训练网络,那么数据仅仅需要图片和图片相对应的label sequence即可。

假设经过recurrent layer后的输出为\mathbf{y}=y_{1}, \cdots, y_{T},每一个y_{t}\in \mathbb{R}^{{L}'}是在集合{L}'=L\cup上概率分布,其中L包含任务中的所有lable和一个空白blank label。在序列\pi \in {L}'^{T}上定义一个sequence-to-sequence的函数映射B,B是将\pi映射到\mathbf{l},通过先移除相同的labels,随后移除空白。那么条件概率则被定义为B将所有\pi映射到\mathbf{l}的概率和p\left( \mathbf{l}| \mathbf{y}\right)=\sum_{\mathbf{\pi}:B\left ( \mathbf{\pi} \right )= \mathbf{l}}p\left ( \mathbf{\pi} | \mathbf{y} \right ) = \sum_{\mathbf{\pi}:B\left ( \mathbf{\pi} \right )= \mathbf{l}} \prod_{t=1}^{T}y_{\pi_{t}}^{t}, 其中y_{\pi_{t}}^{t}是指在t时刻存在label \pi_{t}的概率。

transcription分为lexicon-free transcription 和 lexicon-based transcription,具体计算方法还得阅读其他论文,目前这一块还不是重点,得到实现时再细看。

网络训练(Network Training)

假设训练数据为X=\left \{ \mathbf{I}_{i}, \mathbf{l}_{i} \right \}_{i},其中\mathbf{I}_{i}是训练图片,\mathbf{l}_{i}是ground truth label sequence,目标方程是要最小化条件概率的负log-likelihood,O=-\sum_{\mathbf{I}_{i},\mathbf{l}_{i}\in X}log p\left ( \mathbf{l}_{i}| \mathbf{y}_{i} \right ),其中y_{i}是指CNN和RNN从输入图像\mathbf{I}_{i}计算得来的序列。

网络架构借鉴了VGG架构,输入图片的高度值固定为32,网络训练时采用SGD,对于optimization则使用Adadelta。

图像文字识别初探(一)-CRNN(Convolution Recurrent Neural Network)和DTRN(Deep-text Recurrent Network)_第2张图片

总结:

  • CRNN是端到端的训练网络,直接从序列标签中学习
  • 对识别的对象不限定长度,仅仅需要height normalization
  • 带字典和不带样本的样本都可以

DTRN(Deep-Text Recurrent Network)

论文:Reading Scene Text in Deep Convolutional Sequences

DTRN是来自于深圳先进技术研究院乔宇老师团队,将场景文字识别看作序列标注问题。思想和上述的白翔老师团队的类似,都是充分利用CNN和RNN的特性,CNN提取特征,RNN预测序列,将两者整合到一个统一的网络里,实现端到端的训练。

DTRN总架构如下图所示,整个系统是端到端的,它以一个带有单词的图片作为输入,直接输出相对应的单词序列,不需要预处理和后向处理等那等操作。首先由一个特殊的CNN网络架构,将输入图片编码成一个有顺序的序列(an ordered swquence),然后使用一个RNN网络将CNN输出的序列解码成一个单词串。对于输入的图像和输出的单词串的长度是不加限制的。

图像文字识别初探(一)-CRNN(Convolution Recurrent Neural Network)和DTRN(Deep-text Recurrent Network)_第3张图片

通过在输入图片上滑动子窗口的形式保证在CNN模型下输出一个有序的特征序列。CNN网络架构如下,CNN的输入图片大小为32×32,与滑窗的窗口大小一致,maxout CNN模型有五个卷积层,每个卷积层后面跟随two or four-group Maxout操作。对于输入的图片,将它的高度resize成32,并且保持它原先的纵横比。

图像文字识别初探(一)-CRNN(Convolution Recurrent Neural Network)和DTRN(Deep-text Recurrent Network)_第4张图片

 下面的RNN选择双向LSTM架构,由于LSTM输出的长度与目标单词串的长度不一致,因此引入CTC将LSTM的序列输出匹配到目标字符串,这个和上面的RCNN类似,不再描述。

总结:

  • 端到端训练
  • 可以利用有用的文本信息识别高度模糊的词汇,不需要预处理和后期处理
  • 深度CNN特征足可以抵抗严重扭曲的单词
  • 包含词汇图片中明确的顺序信息,这是划分单词串的根本
  • 不依赖预先定义的词典,可以处理未知单词和任意单词串

你可能感兴趣的:(图像文字识别,深度学习,机器学习,tensorflow,机器学习和深度学习之旅)