开放科学(资源服务)标志码(OSID):
场景文本检测有助于场景内容信息的获取、分析和理解,对于提高图像检索能力、工业自动化水平和场景理解能力等具有重要意义,可应用于自动驾驶、车牌票据识别、智能机器人、图片检索和大数据产业等场景。目前,场景文本检测已成为计算机视觉与模式识别、文档分析与识别领域的研究热点[1-2]。相较于通用目标检测,在同一张或不同自然场景图片中文本尺度变化较大,最大文本与最小文本之间可以相差近230倍[3]。因此,多尺度场景文本检测网络应运而生,多尺度场景文本检测网络通过多尺度多形式的特征提取和融合,以适应场景文本尺度的多变性,对于场景文本检测的工业化应用具有十分重要的意义。
早期多尺度场景文本检测网络采用传统机器学习的方法,通过传统的图像处理方法和人工设计的特征检测场景文本。如文献[4]通过提取最大稳定极值区域(MSER)找出候选字母,根据Hu矩特征刻画候选字母的几何特征;然后通过单链接聚类得到候选文本,最后引入共生纹理特征筛选文本区域,该算法对文本尺度变化有一定的适应性。文献[5]采用基于方向预分类的Gabor小波变换特征提取方法,利用Gabor函数良好的频率选择性和方向选择性,同时考虑到笔画相对位置的偏移,对笔画变形和低分辨率字符具有较好的适应性。文献[6]引入笔画宽度变化(SWT)算法处理场景图片提取不同尺度和不同方向的文本候选区,采用手工特征和随机森林(Random Forest,RF)算法过滤非文本区域,利用文本间的相似性连接成文本行。文献[7]采用多尺度滑动窗口模型,针对文本的局部特征提出一种基于文本部件的树形结构,该算法能很好地适应文本尺度的多变性。上述方法虽然在多尺度场景文本检测领域取得了不错的效果,但是传统的机器学习方法在特征提取方面仍有许多不足:由于场景文本检测的复杂性,人工设计特征难度高,需要消耗大量的时间和人力,成本较高;人工设计的特征会引入人为因素,可能造成文本特征的缺失甚至引入错误特征,检测精度不高。
近年来,随着深度学习网络的迅猛发展,涌现出一系列多尺度场景文本检测网络,其中典型方法是基于金字塔网络,根据其网络结构可分为单向金字塔网络和双向金字塔网络。
基于单向金字塔网络的方法[8-11]通常只有自下而上的特征提取过程,即对原始输入图像通过卷积、池化等操作进行特征提取,在不同层的特征图或者融合后的特征图上进行文本检测。根据网络输入不同大致可分为两类:一类为单向特征金字塔网络,其输入为单一尺度场景图片,在同一图片不同层的特征图或者不同层融合后的特征图上进行文本检测;另一类为单向图片金字塔网络,该类方法输入不同尺度图片,在不同图片的不同尺度特征图上进行检测。如CTPN[8]为解决文本长度变化非常剧烈的问题,采用单向特征金字塔网络,通过设定相同宽度不同高度的垂直文本候选框,使用长短期双向记忆模型处理文字建议序列,进而计算多尺度文字区域外接框与置信度。R2CNN[9]使用Faster R-CNN网络提取特征,利用不同尺寸卷积核的ROI Pooling处理特征图,从而计算出文本目标的矩形包围框检测多尺度场景文本。TextBoxss[10]基于SSD框架,根据不同卷积层的多尺度特征检测不同尺寸文本,通过设定不同纵横比的默认文本候选框,提高了不同尺寸文本的检测准确率。SSTD[11]基于SSD框架,借鉴了GoogleNet[12]中的Inception模块,使用HIM(Hierarchical Inception Module)融合卷积特征以提高模型的性能。
基于单向金字塔结构网络虽然在多尺度场景文本检测领域取得了较好的性能,但是单向特征金字塔网络只包含自下而上的特征提取过程,可以提取到丰富高层语义特征,却忽略了低层特征图包含的文本边界特征信息,造成文本边界检测不准确;而单向图片金字塔网络虽然通过对不同尺度图片进行特征提取,有利于检测不同尺度文本,但是增加了额外的计算开销。因此,针对单向金字塔网络的不足,研究人员提出了基于双向金字塔结构的多尺度场景文本检测网络。
基于双向金字塔网络的方法包括自下而上和自上而下两个特征提取过程,首先在自上而下过程中融合同层次自下而上的特征图,最后在不同尺度特征图或融合后的特征图上进行检测,该类方法可以有效利用低层特征分辨率高和高层特征语义信息丰富的特点,其中低层特征语义信息少但分辨率高,有利于文本目标的定位,而高层特征分辨率低但包含语义信息丰富,有利于文本目标和非文本目标的分类。典型方法为基于FPN[13](Feature Pyramid Network)结构,如PSENet[14]引入FPN结构,首先将文本区域收缩划分为多级中心区域,然后进行像素级分类和预测多级中心区域,最后使用广度优先搜索算法逐级扩展为不同的文本区域,可以有效地检测相距较近和尺度多变的文本实例。MSR[15]网络使用FPN结构,输入多个尺度原始图像进行特征提取并在相同层对不同尺度图片的特征进行融合,在最后融合后的特征图上检测多尺度的文本。CCTN[16]网络基于VGG框架,首先将文本划分为文本区域和文本中心线区域,然后通过先粗略分类后精细分割的方式检测多尺度文本。EAST[17]则基于PVANet,通过上采样融合不同网络层的特征计算出融合特征图,在融合特征图的基础上回归文本包围框的相关属性来检测场景文本。PixelLink[18]采用基于VGG-16的双向特征金字塔网络,通过预测像素类别和像素在空间特征上的连通性以区分不同文本。
基于双向金字塔网络的方法虽然同时融合高层语义特征和低层包含的边界特征,但其在自下而上的深层特征提取过程中,会通过下采样来减小特征图的分辨率以增大感受野来提取高层的语义特征,因此存在以下不足:大文本边界回归弱,虽然较深的特征图有利于预测大文本,但随着特征图分辨率的减小和网络层数的加深,大文本的边界信息也会逐渐减少,因而不利于大文本的边界回归;小文本语义信息丢失。随着特征图分辨率的减小,小文本的语义特征会逐渐减少甚至丢失,导致出现漏检情况;文本边界检测错误,随着大文本边界信息的减少和小文本语义特征的减弱,造成文本检测框包含过多背景或部分包围文本造成的边界检测识别错误,虽然该类方法在自上而下的过程中融合了同层次的自下而上的浅层特征图以增强低层边界特征,但是对于大文本而言,浅层特征图的语义信息弱,没有能力检测大文本,而对于小文本而言,由于其语义特征已经丢失,即使融合特征,检测效果也不会有明显提升。
针对金字塔结构由于下采样减小特征图分辨率而造成的性能次优情况,本文采用多尺度特征权重融合(Multi-Scale Feature Weight Fusion,MSWF)模型在3个分支上进行多尺度特征提取,以兼顾高层语义特征和低层边界特征。由于不同分支特征之间的不一致性,本文在3个分支加入可学习的权重,并针对多尺度场景文本边界检测错误的问题,提出学习主动中心轮廓(Learning Active Center Contour,LACC)模型用于多尺度场景文本边界检测。
1 基于LACC模型的场景文本检测网络 1.1 场景文本检测算法框架本文提出的基于学习主动中心轮廓模型的场景文本检测网络结构如图 1所示。本文网络使用ResNet[19]作为主干网,首先在其基础上构建多尺度特征权重融合(MSWF)模型,在3个不同分支上进行多尺度的特征提取和权重融合,然后借鉴FPN结构提取自上而下特征图,接着使用融合函数
![]() |
Download:
|
图 1 基于学习主动中心轮廓模型的场景文本检测网络 Fig. 1 Scene text detection network based on learning active center contour model |
本文网络特征提取过程如图 1所示,首先将增强后的场景文本图片输入到自下而上的主干网中,依次计算出3个不同尺度的特征图
$ \begin{array}{l}F=C({P}_{2}, {P}_{3}, {P}_{4}, {P}_{5}, {P}_{6})=\\ {}_{}{}_{}{P}_{2}\left|\right|\mathrm{U}\mathrm{p}\left({P}_{3}\right)\left|\right|\mathrm{U}\mathrm{p}\left({P}_{4}\right)\left|\right|\mathrm{U}\mathrm{p}\left({P}_{5}\right)\left|\right|\mathrm{U}\mathrm{p}\left({P}_{6}\right)\end{array} $ | (1) |
其中:
多尺度特征权重融合模型结构如图 2所示,将图 1所示特征图
$ {B}_{1}={W}_{1}\odot {D}_{1} $ | (2) |
![]() |
Download:
|
图 2 多尺度特征权重融合模型示意图 Fig. 2 Schematic diagram of multi-scale feature weight fusion model |
其中:⊙为Hadamard乘积,按照上述流程依次计算出各分支输出特征图(
$ {M}_{\mathrm{M}\mathrm{F}}=\mathrm{c}\mathrm{a}\mathrm{t}({B}_{1}, {B}_{2}, {B}_{3}) $ | (3) |
相较于三叉戟网络[21](Trident Network,TridentNet),本文网络不同之处在于着重对3个分支的融合过程进行改进,TridentNet三分支卷积核参数权值共享,首先对训练目标进行尺度划分,然后根据划分结果选择分支进行训练,最后使用第二分支进行推理,并经过NMS后处理输出推理结果。
因此,MSWF Model与TridentNet存在以下不同之处:1)TridentNet对训练目标进行尺度划分来选择训练分支,三分支卷积核参数权值共享,而本文采用三分支并行训练,分别计算三分支权重;2)TridentNet通过第二分支进行检测,本文在权重融合后的特征图上检测文本;3)TridentNet选择某个分支进行训练和检测属于硬注意力的一种,本文对三分支的特征基于权重融合属于软注意力。
MSWF模型各分支的空洞卷积[22]分别使用了不同空洞率(Dilate=1,Dilate=2,Dilate=3),采用三分支结构是为了在多个分支提取多种尺度特征,以适应场景文本的多尺度变化。相同分支则在同一尺度采用空洞卷积,既可以保持分辨率不变,又可以扩大感受野,以此取代特征金字塔下采样提取文本特征的方法,提高了网络对于大、小文本的检测性能。
图 2中权重计算模块结构如图 3所示,
![]() |
Download:
|
图 3 权重计算模块结构 Fig. 3 Structure of the weight calculation block |
在具体计算权重时,令
$ {\boldsymbol{y}}_{ij}^{}={\boldsymbol{\alpha }}_{ij}\cdot {\boldsymbol{x}}_{ij}^{1}+{\boldsymbol{\beta }}_{ij}^{}\cdot {\boldsymbol{x}}_{ij}^{2}+{\boldsymbol{\gamma }}_{ij}^{}\cdot {\boldsymbol{x}}_{ij}^{3} $ | (4) |
其中:
$ {\boldsymbol{\alpha }}_{ij}=\frac{{\mathrm{e}}^{{\lambda }_{{\boldsymbol{\alpha }}_{ij}}}}{{\mathrm{e}}^{{\lambda }_{{\boldsymbol{\alpha }}_{ij}}}+{\mathrm{e}}^{{\lambda }_{{\boldsymbol{\beta }}_{ij}}}+{\mathrm{e}}^{{\lambda }_{{\boldsymbol{\gamma }}_{ij}}}} $ | (5) |
其中:
算法1 MSWF算法
输入 ResNet50第三层输出卷积特征图
输出 多尺度特征权重融合特征图
步骤1 将
步骤2 使用
步骤3 将
步骤4 重复步骤2和步骤3,依次计算出卷积特征图(
步骤5 重复步骤4依次计算出权重特征图(
步骤6 使用cat函数拼接(
步骤7 将Wt输入到softmax函数归一化处理,计算出各分支权重特征图(
步骤8 计算权重特征图(
步骤9 使用cat函数拼接特征图(
在众多分割方法中,基于主动轮廓模型(ACM)或其变形模型是图像分割中应用最广泛的方法之一,取得了较好的性能。本文提出的学习主动中心轮廓模型即是受主动轮廓模型的启发。1988年,KASS等[24]提出主动轮廓模型,将图像分割问题转换为求解能量泛函最小值问题,为图像分割提供一种全新的思路。该模型的主要原理是通过构造能量泛函,在能量函数最小值驱动下,使用基于偏微分的方法最小化能量函数,使轮廓线朝着目标边界的方向不断演进,最终分割出目标。但由于实际图像的背景是不均匀的,并且背景和目标的对比度往往比较低,仅依靠能量函数的最小值无法准确分割出目标,因此Chan-Vese模型将曲线所围的面积和曲线长度作为能量项引入到能量函数中。在过去若干年里,研究人员提出了很多基于ACM的模型,如无边缘主动轮廓模型(ACWE)和BRESSON等[25]提出的快速全局最小化主动轮廓模型(FGM-ACM)。
ACWE模型的能量最小化问题可以表述为:
$ \underset{{{\mathit{\Omega}} }_{c}, {c}_{1}, {c}_{2}}{\mathrm{m}\mathrm{i}\mathrm{n}}\left\{{E}_{1}^{\mathrm{A}\mathrm{C}\mathrm{W}\mathrm{E}}({{\mathit{\Omega}} }_{c}, {c}_{1}, {c}_{2}, \lambda )\right.={\int }_{0}^{\mathrm{L}\mathrm{e}\mathrm{n}\mathrm{g}\mathrm{t}\mathrm{h}\left(C\right)}ds+ \\ \lambda {\int }_{{\mathit{\Omega}} }({c}_{1}-f{\left(x\right))}^{2}\mathrm{d}x+\lambda {\int }_{{\mathit{\Omega}} /{{\mathit{\Omega}} }_{c}}({c}_{2}-f{\left(x\right))}^{2}\left.\stackrel{}{\mathrm{d}x}\right\} $ | (6) |
其中:
虽然基于主动轮廓模型的图像分割取得了很好的效果,但还存在以下不足:1)采用无监督的方法不需要从训练数据中学习属性,因此它们很难处理噪声和遮挡;2)有许多参数是依据经验设定的;3)多数方法都不能对自然场景图片进行鲁棒分割。显然,基于主动轮廓模型的方法不适用于有监督的机器学习来处理标记图像,而且大多数基于深度学习的自然场景文本检测方法缺乏整合目标先验知识的机制。因此,有必要将基于深度学习的场景文本检测与基于主动轮廓模型的方法结合起来,以便后者能够提供足够的先验知识,以提高模型对于场景文本边界的检测性能。
本文受主动轮廓模型ACWE模型能量最小化问题的启发,提出一个整合了中心点、文本区域和文本检测框长度信息的可用于深度神经网络学习的新损失函数,将LACC Mode的先验知识用于神经网络训练,解决多尺度场景文本检测框轮廓线能量全局最小化问题,进而精确检测文本边界。具体的损失函数如式(7)所示:
$ \mathrm{L}\mathrm{o}\mathrm{s}{\mathrm{s}}_{\mathrm{L}\mathrm{A}\mathrm{C}\mathrm{C}}={\lambda }_{c}\mathrm{c}\mathrm{e}\mathrm{n}\mathrm{t}\mathrm{e}\mathrm{r}+{\lambda }_{l}\mathrm{l}\mathrm{e}\mathrm{n}\mathrm{g}\mathrm{t}\mathrm{h}+{\lambda }_{r}\mathrm{r}\mathrm{e}\mathrm{g}\mathrm{i}\mathrm{o}\mathrm{n} $ | (7) |
其中:
$ \begin{array}{l}{c}_{\mathrm{c}\mathrm{e}\mathrm{n}\mathrm{t}\mathrm{e}\mathrm{r}}=\frac{-1}{N}\sum \limits_{xyc}\left\{\begin{array}{l}(1-{\widehat{Y}}_{xyc}{)}^{a}\mathrm{l}\mathrm{o}{\mathrm{g}}_{a}({\widehat{Y}}_{xyc}), {Y}_{xyc}=1\\ (1-{Y}_{xyc}{)}^{\beta }\mathrm{l}\mathrm{o}{\mathrm{g}}_{a}({\widehat{Y}}_{xyc}{)}^{\alpha }, \mathrm{其}\mathrm{他}\\ \mathrm{l}\mathrm{o}{\mathrm{g}}_{a}(1-{\widehat{Y}}_{xyc})\end{array}\right.\\ {l}_{\mathrm{l}\mathrm{e}\mathrm{n}\mathrm{g}\mathrm{t}\mathrm{h}}=\sum\limits _{{\mathit{\Omega}} }^{i=1, j=1}\sqrt{\left|(\nabla {u}_{{x}_{i, j}}{)}^{2}+(\nabla {u}_{{y}_{i, j}}{)}^{2}\right|+\varepsilon }\\ {r}_{\mathrm{r}\mathrm{e}\mathrm{g}\mathrm{i}\mathrm{o}\mathrm{n}}=\left|\sum\limits _{{\mathit{\Omega}} }^{i=1, j=1}{u}_{i, j}(c{}_{1}-{v}_{i, j}{)}^{2}\right|+\\ \left|\sum \limits_{{\mathit{\Omega}} }^{i=1, j=1}(1-{u}_{i, j})(c{}_{2}-{v}_{i, j}{)}^{2}\right|\end{array} $ | (8) |
其中:center中心点表示矩形文本框的中心点,采用Focal Loss [26]损失函数;
$ \begin{array}{l}{c}_{1}=\int \nu \cdot \mu \mathrm{d}x/\int \mu \mathrm{d}x\\ {c}_{2}=\int \nu \cdot (1-\mu )\mathrm{d}x/\int (1-\mu )\mathrm{d}x\end{array} $ | (9) |
综上,本文提出一种新损失函数,考虑了文本的中心点和边界轮廓的长度以及文本区域的拟合度,具有以下优点:1)将无监督AC Model能量最小化问题转化为有监督的深度学习损失函数最小化问题;2)将本文网络提取特征和AC Model的先验知识相结合,解决了AC Model过于依赖人工特征设计、鲁棒性差和深度学习缺乏足够先验知识的问题;3)将AC Model基于图像像素信息检测方式转化为基于深度学习卷积特征图像素信息检测方式。
2.3 学习主动中心轮廓模型算法算法2 LACC算法
输入 本文网络输出特征图
输出 网络损失
步骤1 对特征图
步骤2 初始化
步骤3 判断
步骤4 如果
$ {c}_{\mathrm{c}\mathrm{e}\mathrm{n}\mathrm{t}\mathrm{e}\mathrm{r}}=\frac{-1}{N}\sum\limits _{xyc}\left(1\right.-{\widehat{Y}}_{xyc}{)}^{a}lo{g}_{a}\left({\widehat{Y}}_{xyc}\right) $ |
其中:
步骤5 如果
$ {c}_{\mathrm{c}\mathrm{e}\mathrm{n}\mathrm{t}\mathrm{e}\mathrm{r}}=\frac{-1}{N}\sum\limits _{xyc}\left(1\right.-{Y}_{xyc}{)}^{\beta }lo{g}_{a}({\widehat{Y}}_{xyc}{)}^{\alpha }lo{g}_{a}\left(1\right.-{\widehat{Y}}_{xyc}) $ |
步骤6 计算文本框的长度损失length,具体公式见式(8)。
步骤7 计算文本框区域损失region,具体公式见式(8)。
步骤8 计算网络损失函数
步骤9 得到学习主动中心轮廓模型算法损失
MSRA-TD500[6]是一个文本尺度变化较大的中英文数据集,共包含500张图片,其中300张用于训练,200张用于测试。
ICDAR-2013(IC13)[28]是一种常用的多方向英文场景文本检测数据集,共包含509张图片,其中258张图片用于训练,251张图片用于测试。文本区域是由四边形的左上、右下2个顶点标注。
ICDAR-2015(IC15)[29]是一种常用的多方向英文文本检测数据集,共包含1 500张图片,其中1 000张图片用于训练,500张图片用于测试。文本区域是由四边形的4个顶点标注。
ICDAR-2017MLT(IC17-MLT)[30]是一个大规模的多方向多语言场景文本数据集,包括7 200张训练图片、1 800张验证图片和9 000张测试图片,由9种自然语言组成,文本区域由四边形的4个顶点标注。
3.2 实验参数本文使用ResNet50[19]在ImageNet[31]上的预训练模型,并在其基础上构建多尺度特征权重融合模型(MSWF Model)作为网络的主干网,所有网络都采用Adam[32]优化器进行训练。首先在IC17-MLT数据集上进行训练,得出IC17-MLT上的训练模型并进行测试,然后加载该训练模型为预训练模型,在MSRA-TD500、IC13和IC15数据上分别继续训练网络并进行测试。本文实验在GTX1080Ti
在训练时,忽略数据集中标注为“不用关注”的模糊文本区域。本文根据实验结果,将损失函数的权重系数设定为:
本文使用准确率(P)、召回率(R)和F-measure(F)对算法进行评估,具体定义如下:
$ P=\frac{{T}_{\mathrm{T}\mathrm{P}}}{{T}_{\mathrm{T}\mathrm{P}}+{F}_{\mathrm{F}\mathrm{P}}} $ | (10) |
$ R=\frac{{T}_{\mathrm{T}\mathrm{P}}}{{T}_{\mathrm{T}\mathrm{P}}+{F}_{\mathrm{F}\mathrm{N}}} $ | (11) |
$ F=2\times \frac{P\times R}{P+R} $ | (12) |
其中:TP表示将正样本预测为正样本数目;FP表示将负样本预测为正样本的误报数;FN表示将正样本预测为负样本的漏报数目。准确率(P)与召回率(R)之间可能会出现矛盾的情况,其中一个测试指标较高,而另外一个测试指标较低。这时就需要综合考虑两者指标的情况,采取F-measure评估方法。
3.4 消融实验 3.4.1 多尺度特征权重融合模型的有效性为验证多尺度特征权重融合模型对本文多尺度场景文字检测网络性能的影响,保持主干网为ResNet50不变,通过对网络添加和去除多尺度特征权重融合模型分别进行训练。本文实验在数据集IC13和IC15上分别进行了测试。实验结果如表 1所示(粗体表示最优值),在IC13数据集上添加多尺度特征权重融合模型,相较于ResNet50+FPN网络F值提升2%,相较于TridentNet网络F值提升1%。在IC15数据集上添加多尺度特征权重融合模型(MSWF Model),相较于ResNet50+FPN网络F值提升2%,相较于TridentNet网络F值提升1%。实验结果表明多尺度特征权重融合模型有效提升了网络的检测性能。
![]() |
下载CSV 表 1 多尺度特征权重融合模型的消融实验结果 Table 1 Ablation experiment result of multi-scale feature weight fusion model |
研究实验结果证明,更优损失函数的使用可以提高大规模图像分类和目标检测的性能。为了更好地验证本文提出网络的检测能力,本文实验通过使用不同的损失函数,在MSRA-TD500数据集上进行测试来验证学习主动中心轮廓模型(LACC Model)对于网络检测能力的影响。保持网络结构不变,使用不同的损失函数分别进行实验,实验结果如表 2所示(粗体表示最优值),使用学习主动中心轮廓模型相较于L1 Loss和SmoothL1 Loss,F值分别提升4%和2%,说明本文提出的学习主动中心轮廓模型可以更好地检测场景文本的边界,提高检测性能。
![]() |
下载CSV 表 2 学习主动中心轮廓模型的消融实验结果 Table 2 Ablation experiment result of learning active center contour model |
本文主要进行了以下方面的实验:
1)多尺度文本检测实验。本文在MSRA-TD500数据集上对本网络进行了测试,以验证其检测多尺度文本的能力。本实验分别在GTX1080Ti
![]() |
下载CSV 表 3 在MSRA-TD500数据集上的检测结果 Table 3 Detection results on MSRA-TD500 dataset |
2)多方向英文文本检测实验。本文在IC13和IC15数据集上对本文方法进行了测试,以验证其检测多方向英文文本的能力。实验采用本文在IC17MLT上的训练模型作为预训练模型,分别在GTX1080Ti
![]() |
下载CSV 表 4 在IC13和IC15数据集上的检测结果 Table 4 Detection results on IC13 and IC15 dataset |
3)多方向多语言文本检测实验。为了测试本文方法对多方向多语言场景文本检测的鲁棒性,本文实验在IC17-MLT基准数据集上对其进行了评估。在IC17MLT上使用GTX1080Ti
![]() |
下载CSV 表 5 在IC17MLT数据集上的测试结果 Table 5 Detection results on IC17MLT dataset |
表 6为本文方法与基于双向特征金字塔结构的PixelLink和三叉戟方法TridentNet在各数据集上代表性检测结果。从表 6可以看出:PixelLink和TridentNet方法对于某些大尺度的文本目标存在欠检测,即大尺度文本检测框不能完全包围目标,小尺度文本目标漏检的情况;而本文方法采用多尺度权重融合与学习主动中心轮廓模型相结合的方式,对于大尺度文本检测效果更好,检测框包围更准确,进一步提高小目标检测能力,有利于解决小目标漏检的问题。
![]() |
下载CSV 表 6 不同方法在各数据集上的代表性检测结果 Table 6 Representative detect results of different methods on each dataset |
本文针对场景文本多尺度变化造成的小文本漏检、大文本欠检测以及场景文本边界检测错误问题,提出基于学习主动中心轮廓模型的场景文本检测网络。通过多尺度特征权重融合模型解决多尺度场景文本特征提取问题,基于学习主动中心轮廓模型解决场景文本边界检测错误的问题,并在4个公共数据集上验证了本文网络对于多尺度场景文本检测的有效性。下一步拟将本文提出网络用于弯曲场景文本检测,以提高网络的泛化性能,并研究本文网络对于类文本元素的检测能力,以增强网络的鲁棒性。
[1] |
姜维, 张重生, 殷绪成. 基于深度学习的场景文字检测综述[J]. 电子学报, 2019, 47(5): 1152-1161. JIANG W, ZHANG C S, YIN X C. Deep learning based scene text detection: a survey[J]. Acta Electronica Sinica, 2019, 47(5): 1152-1161. (in Chinese) |
[2] |
王建新, 王子亚, 田萱. 基于深度学习的自然场景文本检测与识别综述[J]. 软件学报, 2020, 31(5): 1465-1496. WANG J X, WANG Z Y, TIAN X. Review of natural scene text detection and recognition based on deep learning[J]. Journal of Software, 2020, 31(5): 1465-1496. (in Chinese) |
[3] |
SINGH B, DAVIS L S. An analysis of scale invariance in object detection-SNIP[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2018: 3578-3587.
|
[4] |
杨玲玲, 叶东毅. 一种基于图像矩和纹理特征的自然场景文本检测算法[J]. 小型微型计算机系统, 2016, 37(6): 1313-1317. YANG L L, YE D Y. Moment and texture based algorithm for text detection in natural scene Images[J]. Journal of Chinese Computer Systems, 2016, 37(6): 1313-1317. (in Chinese) |
[5] |
尹芳, 陈德运, 吴锐. 改进的Gabor小波变换特征提取方法[J]. 计算机工程, 2012, 38(15): 145-147. YIN F, CHEN D Y, WU R. Improved Gabor wavelet transformation feature extraction method[J]. Computer Engineering, 2012, 38(15): 145-147. (in Chinese) |
[6] |
YAO C, BAI X, LIU W Y, et al. Detecting texts of arbitrary orientations in natural images[C]//Proceedings of 2012 IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2012: 1083-1090.
|
[7] |
SHI C Z, WANG C H, XIAO B H, et al. Scene text recognition using part-based tree-structured character detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2013: 2961-2968.
|
[8] |
TIAN Z, HUANG W L, HE T, et al. Detecting text in natural image with connectionist text proposal network[C]//Proceedings of European Conference on Computer Vision. Berlin, Germany: Springer, 2016: 56-72.
|
[9] |
JIANG Y Y, ZHU X Y, WANG X B, et al. R2CNN: rotational region CNN for orientation robust scene text detection[EB/OL]. [2021-01-05]. https://arxivpreprintarxiv:1706.09579.
|
[10] |
LIAO M H, SHI B G, BAI X, et al. TextBoxes: a fast text detector with a single deep neural network[C]//Proceedings of AAAI Conference on Artificial Intelligence. [S. 1. ]: AAAI Press, 2017: 235-136.
|
[11] |
HE P, HUANG W L, HE T, et al. Single shot text detector with regional attention[C]//Proceedings of IEEE International Conference on Computer Vision. Washington D.C., USA: IEEE Press, 2017: 3047-3055.
|
[12] |
SZEGEDY C, LIU W, JIA Y, et al. Going deeper with convolutions[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2015: 1-9.
|
[13] |
LIN T Y, DOLLAR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2017: 2117-2125.
|
[14] |
WANG W H, XIE E Z, LI X, et al. Shape robust text detection with progressive scale expansion network[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2019: 9336-9345.
|
[15] |
XUE C H, LU S J, ZHANG W. MSR: multi-scale shape regression for scene text detection[EB/OL]. [2021-01-05]. https://arxiv preprint arxiv: 1901.02596.
|
[16] |
HE T, HUANG W L, QIAO Y, et al. Accurate text localization in natural image with cascaded convolutional text network[EB/OL]. [2021-01-05]. https://arxivpreprintarxiv:1603.09423.
|
[17] |
ZHOU X Y, YAO C, WEN H, et al. EAST: an efficient and accurate scene text detector[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2017: 2642-2651.
|
[18] |
DENG D, LIU H F, LI X L, et al. Pixellink: detecting scene text via instance segmentation[C]//Proceedings of AAAI Conference on Artificial Intelligence. [S. 1. ]: AAAI Press, 2018: 358-367.
|
[19] |
HE K M, ZHANG X Y, REN S Q, et al. Identity mappings in deep residual networks[C]//Proceedings of European Conference on Computer Vision. Berlin, Germany: Springer, 2016: 630-645.
|
[20] |
ZHU X, HU H, LIN S, et al. Deformable ConvNets V2: more deformable, better results[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2019: 9300-9308.
|
[21] |
LI Y H, CHEN Y T, WANG N Y, et al. Scale-aware trident networks for object detection[C]//Proceedings of IEEE/CVF International Conference on Computer Vision. Washington D.C., USA: IEEE Press, 2019: 6053-6062.
|
[22] |
YU F, KOLTUN V. Multi-scale context aggregation by dilated convolutions[EB/OL]. [2021-01-05]. https://arxivpreprintarxiv:1511.07122.
|
[23] |
WANG G R, WANG K, LIN L. Adaptively connected neural networks[C]//Proceedings of IEEE/CVF Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2019: 1781-1790.
|
[24] |
KASS M, WITKIN A, TERZOPOULOS D. Snakes: active contour models[J]. International Journal of Computer Vision, 1988, 1(4): 321-331. DOI:10.1007/BF00133570 |
[25] |
BRESSON X, ESEDOGLU S, VANDERGHEYNST P, et al. Fast global minimization of the active contour/snake model[J]. Journal of Mathematical Imaging and Vision, 2007, 28(2): 151-167. DOI:10.1007/s10851-007-0002-0 |
[26] |
LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection[C]//Proceedings of IEEE International Conference on Computer Vision. Washington D.C., USA: IEEE Press, 2017: 2999-3007.
|
[27] |
LAW H, DENG J. CornerNet: Detecting objects as paired keypoints[C]//Proceedings of European Conference on Computer Vision. Washington D.C., USA: IEEE Press, 2018: 734-750.
|
[28] |
KARATZAS D, SHAFAIT F, UCHIDA S, et al. ICDAR 2013 robust reading competition[C]//Proceedings of the 12th International Conference on Document Analysis and Recognition. Washington D.C., USA: IEEE Press, 2013: 1484-1493.
|
[29] |
KARATZAX D, GOMEZ-BIGORDA L, NICOLAOU A, et al. ICDAR 2015 competition on robust reading[C]//Proceedings of the 13th International Conference on Document Analysis and Recognition. Washington D.C., USA: IEEE Press, 2015: 1156-1160.
|
[30] |
NAYEF N, YIN F, BIZID I, et al. ICDAR 2017 robust reading challenge on multi-lingual scene text detection and script identification-RRC-MLT[C]//Proceedings of the 14th IAPR International Conference on Document Analysis and Recognition. Washington D.C., USA: IEEE Press, 2017: 1454-1459.
|
[31] |
DENG J, DONG W, SOCHIR R, et al. ImageNet: a large-scale hierarchical image database[C]//Proceedings of 2009 IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2009: 248-255.
|
[32] |
KINGMA D P, BA J. Adam: a method for stochastic optimization[EB/OL]. [2021-01-05]. https://arxivpreprintarxiv:1412.6980.
|
[33] |
REN S, HE K, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[EB/OL]. [2021-01-05]. https://arxivpreprintarxiv:1506.01497.
|
[34] |
SHI B, BAI X, BELONGIE S. Detecting oriented text in natural images by linking segments[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2017: 3482-3490.
|
[35] |
LONG S B, RUAN J Q, ZHANG W J, et al. TextSnake: a flexible representation for detecting text of arbitrary shapes[C]//Proceedings of European Conference on Computer Vision. Berlin, Germany: Springer, 2018: 20-36.
|
[36] |
LIAO M H, ZHU Z, SHI B G, et al. Rotation-sensitive regression for oriented scene text detection[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2018: 5909-5918.
|
[37] |
LYU P, YAO C, WU W, et al. Multi-oriented scene text detection via corner localization and region segmentation[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2018: 7553-7563.
|
[38] |
LIU X B, LIANG D, YAN S, et al. FOTS: fast oriented text spotting with a unified network[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Washington D.C., USA: IEEE Press, 2018: 5676-5685.
|
[39] |
ZHONG Z Y, SUN L, HUO Q. An anchor-free region proposal network for Faster R-CNN-based text detection approaches[J]. International Journal on Document Analysis and Recognition, 2019, 22(3): 315-327. DOI:10.1007/s10032-019-00335-y |