文本检测算法----TextFuseNet(IJCAI-PRICAI-20)
多种文本检测算法性能对比及算法介绍
(https://blog.csdn.net/qq_39707285/article/details/108754444)
TextFuseNet: Scene Text Detection with Richer Fused Features
- 前言
- 1. 算法简介
- 2. 算法详解
- 2.1 网络结构
- 2.2 Multi-level Feature Representation
- 2.3 Multi-path Fusion Architecture
- 3.4 loss函数
- 3.5 Weakly Supervised Learning
- 4. 测试结果
开源代码地址
前言
自然场景中任意形状文本检测是一项极具挑战性的任务,与现有的仅基于有限特征表示感知文本的文本检测方法不同,本文提出了一种新的框架,即 TextFuseNet ,以利用融合的更丰富的特征进行文本检测。
该算法用三个层次的特征来表示文本,字符、单词和全局级别,然后引入一种新的文本融合技术融合这些特征,来帮助实现鲁棒的任意文本检测。另外提出了一个弱监督学习机制,可以生成字符级别的标注,在缺乏字符级注释的数据集情况下也可以进行训练。
该算法在ICDAR2013上取得F1分数94.3%,在ICDAR2015上F1分数92.1%,在Total-Text上87.1%,在CTW-1500上86.6%,目前为止最佳成绩。
1. 算法简介
之前的文本检测算法大致分为两种,基于字符级别的检测和基于单词级别的检测。基于字符级别的检测算法首先提取单个字符,然后再使用字符合并算法合并这些字符成一个单词,然而这种方法因为要生成大量的字符候选框并且要合并,比较耗时。相比之下,基于单词级别的检测算法直接检测单词,会更高效和简单,但这种方法通常无法有效地检测具有任意形状的文本。为了解决这个问题,一些基于单词的方法进一步应用实例分割来进行文本检测。 在这些方法中,前景分割掩码被估计以帮助确定各种文本形状。
尽管有很好的结果,但现有的基于实例分割的方法仍然有两个主要的局限性。一是,这些方法只基于单个感兴趣区域(RoI)检测文本,而不考虑全局上下文,因此它们倾向于基于有限的视觉信息产生不准确的检测结果。二是,现有的方法没有对不同层次的单词语义进行建模,产生假阳性的可能性增大。从图一中可以看到这种方法的弊端。
本文提出的TextFuseNet能够有效的解决这些问题,并且可以高效准确的预测任意形状的文本。TextFuseNet与其他算法相比主要的区别在于,有效的利用各种层次的特征,例如字符级别的、单词级别的、全局级别的特征,而其他的文本检测算法往往只使用一种层次的特征。
TextFuseNet网络结构主要分为三个分支:
第一个是语义分割分支( semantic segmentation branch),该分支用来提取液全局级别的特征;
另外二个是检测分支和mask分支(detection and mask branches),用来提取字符级别和单词级别的特征;
在得到三种层次的特征后,使用多路径特征融合体系结构(Multi-path Fusion Architecture),融合三者特征,生成更具代表性的特征表示,从而产生更准确的文本检测结果。
目前大部分数据集只包含单词级别的标注,很少有字符级别的标注,为解决字符级别标注数据集缺乏的问题,提出了一种弱监督学习方案,通过从单词级注释数据集学习来生成字符级注释。
总体的结构如图2所示。
2. 算法详解
2.1 网络结构
具体网络结构如图2所示,首先提取多层次的特征,然后执行多路径融合以进行文本检测。该结构主要由5部分组成,
- 使用特征金字塔(FPN)作为backbone进行多特征提取;
- 使用RPN生成文本候选框;
- 语义分割分支生成全局语义特征;
- 检测分支预测单词和字符;
- mask分支生成单词和字符的实例分割;
在TextFuseNet中,使用ResNet作为backbone,RPN生成的文本候选框作为检测和mask分支的输入,在语义分割分支来对输入图像进行语义分割,并帮助获得全局级别的特征。
mask在检测分支中,通过预测文本候选框的类别和采用边界框回归来细化文本候选框,提取和融合了单词和全局级别的特征来检测单词和字符。
mask分支,对从检测分支检测到的对象执行实例分割;
提取和融合所有字符、单词和全局级别的特征,以完成实例分割以及最后的文本检测任务。
2.2节来主要来讲解提取多层次的特征表示,在提取多特征后,多路径融合体系结构来融合不同的特征,用于检测任意形状的文本,多径融合体系结构可以有效地对多层特征进行对齐和合并,以提供健壮的文本检测,多路径融合体系结构的实现细节在2.3节中描述。
2.2 Multi-level Feature Representation
在检测器的检测和掩码分支中,通过预测文本候选框中的字符和单词,能够很容易的获得字符级别和单词级别的特征。 这里应用RoIAlign提取不同的特征,并对单词和字符进行检测。
除了字符和单词特征,还要获取全局的语义特征,如图2所示,语义分割分支是基于FPN的输出构建的。将所有特征层的特征融合到一个统一的特征表示中,并在这个统一的特征表示上执行分割,从而获得全局分段的文本检测结果。通常,使用1×1的卷积将不同特征层的特征的通道数对齐,并将特征映射调整为相同的大小,以便以后统一。
2.3 Multi-path Fusion Architecture
在获取到多级特征后,分别在检测和mask分支采用多径融合体系结构。
在检测分支中,基于从RPN获得的文本候选,提取全局和单词级特征,用于不同路径的文本检测。然后,融合这两种类型的特征,以单词和字符的形式提供文本检测。值得注意的是,在检测分支的时候,不能提取和融合字符级别的特征,因为,在执行检测之前,字符尚未被识别。在实际代码中,给定一个文本候选框,使用RoIAlign从FPN的输出特征中提取到一个大小为7×7的全局和单词特征。使用 element-wise相加融合这两个特征,然后再经过一个3×3的卷积层和一个1×1的卷积层,最终融合后的特征用于分类和坐标回归。
在mask分支,对于每个单词级实例,可以在多路径融合体系结构中融合得到相应的字符、单词和全局级别特征。图3详细说明了多路径融合结构。
在所提出的体系结构中,从不同的路径中提取多层次特征,并将它们融合起来,以获得更丰富的特征,以帮助学习更具鉴别性的特征表示。
给定一个由rir_iri表示的输入单词,首先根据其与字符在字符区域上的交集的比率来识别属于这个词建议的字符结果CiC_iCi,这意味着如果单词框完全覆盖字符,则该比率为1,如果不覆盖字符,则为0。使用cjc_jcj来表示字符,属于单词rir_iri的字符集合CiC_iCi可以表示为:
其中bib_ibi和bjb_jbj分别是单词rir_iri和字符实例cjc_jcj的包围框,T是阈值,将T设为0.8。
这个公式中cic_ici应该是写错了应该是cjc_jcj,这个公式的解释的比较麻烦,根据个人理解简单画了一下,方便理解,如下图所示:
每个单词rir_iri中字符的数量不是固定的,从0到数百不等,所以需要将集合CiC_iCi中字符的特征融合为一个统一的表示。在代码中,首先使用RoIAlign提取CiC_iCi中每个字符的大小为14×14的相应特征,然后通过 element-wise求和融合这些特征映射。再通过一个3×3卷积层和一个1×1卷积可以得到最终的字符级特征。
通过进一步应用RoIAlign提取单词的特征和相应的全局语义特征,通过element-wise求和将这三个层次的特征融合起来,然后通过一个3×3卷积层和一个1×1卷积层去获得更丰富的特征。最后融合的特征用于实例分割。
3.4 loss函数
TextFuseNet总体损失函数由四部分组成,其中LrpnL_{rpn}Lrpn、LsegL_{seg}Lseg、LdetL_{det}Ldet和LmaskL_{mask}Lmask分别是RPN、语义分割分支、检测分支和掩码分支的loss函数。
3.5 Weakly Supervised Learning
TextFuseNet需要检测单词和字符文本,所以就需要字符级别标注的数据集,但目前大部分数据集都只是单词级别的标注,去标注字符级别的文本,十分耗时。于是本文提出一种弱监督学习机制去训练模型。
首先在一个完整的包含单词级别和字符级别的数据上预训练模型MMM,然后对于一个新的只包含单词级别的数据集AAA,通过预训练模型MMM去生成字符训练样本。
具体的,用模型MMM去预测数据集AAA,对于数据集AAA中每张图片,可以得到一组字符候选样本:
这里cic_ici,sis_isi,bib_ibi和mim_imi分别代表预测的第i个字符的类别,置信度分数,bbox和mask,rir_iri代表各个字符。然后通过增加置信度阈值和弱监督的词级注释,在所有生成的字符候选样本中去掉假阳样本,最后得到的正样本如下:
CCC代表所有检出的字符类别,SSS是区分正样本的置信度阈值,(mi⋂gi)mi\frac{(m_i\bigcap g_i)}{m_i}mi(mi⋂gi)是每个候选字符样本rir_iri和它所属的单词级别的GT gig_igi的交并比,TTT是决定候选字符是否在单词内的阈值。由于只有单词级标注的限制,阈值SSS设置的相对较低,有利于保持字符样本的多样性,在代码中SSS和TTT分别设置为0.1和0.8。最后,确定的正字符样本可以作为字符级标注,并与单词级标注相结合,以训练更鲁棒和更准确的文本检测模型。
4. 测试结果
文本检测算法----TextFuseNet(IJCAI-PRICAI-20)相关推荐
- 2020最新文本检测算法TextFuseNet
向AI转型的程序员都关注了这个号???????????? 人工智能大数据与深度学习 公众号:datayx TextFuseNet: Scene Text Detection with Richer ...
- 文本检测算法性能对比
文本检测算法性能对比 1. 任意四边形文本数据集 1.1 ICDAR15[^1] 2. 曲形文本数据集 2.1 CTW1500[^10] 2.2 Total-Text[^11] 本文会一直更新经典的. ...
- 分割文本_PSENet、PANNet、DBNet三个文本检测算法异同
点击蓝字关注我们 这三个文本检测算法都是segment base算法,通过由下而上的方式,先对text进行segment,然后再根据segment text,计算出text的instance PSEN ...
- CVPR 2019 | 文本检测算法PSENet解读与开源实现
点击我爱计算机视觉标星,更快获取CVML新技术 作者:刘恒 编辑:CV君 PSENet文本检测算法来自论文<Shape Robust Text Detection with Progressiv ...
- 文本检测算法新思路:基于区域重组的文本检测
摘要:文本检测是文本读取识别的第一步,对后续的文本识别有着重大的影响.本文主要介绍基于区域重组的文本检测算法. 本文分享自华为云社区<技术综述十三:弯曲文本检测算法(一)>,作者: 我想静 ...
- 4种基于像素分割的文本检测算法
摘要:文本检测是文本读取识别的第一步,对后续的文本识别有着重大的影响.一般场景下,可以通过对通用目标检测算法进行配置修改,来实现对文本行的检测定位.本文主要介绍基于像素分割的文本检测算法. 本文分享自 ...
- PSENet PANNet DBNet 三个文本检测算法异同
向AI转型的程序员都关注了这个号???????????? 机器学习AI算法工程 公众号:datayx 这三个文本检测算法都是segment base算法,通过由下而上的方式,先对text进行seg ...
- AI实战:基于AdvancedEAST的自然场景图像文本检测算法
向AI转型的程序员都关注了这个号???????????? 机器学习AI算法工程 公众号:datayx EAST文本检测与Keras实现 https://blog.csdn.net/linchuha ...
- pse文本行检测_CVPR 2019 | 文本检测算法PSENet解读与开源实现
作者:刘恒 编辑:CV君 PSENet文本检测算法来自论文<Shape Robust Text Detection with Progressive Scale Expansion Networ ...
最新文章
- Java5线程并发库之保障变量的原子性操作
- SQL server 存储过程的建立和调用
- 使用MyBatis集成阿里巴巴druid连接池(不使用spring)
- python网址太长_【原创】分享一个自己用python写的长网址转短网址工具,给需要的你...
- (54)FPGA条件选择有优先级(if-else)
- 微信小程序开发学习笔记001--认识微信小程序,第一个微信小程序
- asp.net gridview 模板列 弹出窗口编辑_如何使用极速PDF编辑器的注释工具?
- Caffe 议事(一):从零开始搭建 ResNet 之 残差网络结构介绍和数据准备
- [转载] LINUS:利用二级指针删除单向链表
- 用户可以通过软件对计算机,用户可以通过____软件对计算机软、硬件资源进行管理。...
- win7显示文件扩展名
- 计算机安全常用防护策略,新手必看
- java嵌套条件运算符_java条件运算符的嵌套使用
- 线性规划与单纯形法(线性规划、单纯形法、单纯形表、人工变量法)
- 2050热身赛 1001 赶火车
- Python绘制论文曲线图
- 明明已经引入了jquery,还是不断地报错,终于找到原因了
- Bit(位)与Byte(字节)的区别
- XCTF攻防世界 Normal_RSA
- LED大屏千兆光纤收发器利亚德洲明LED显示屏诺瓦灵星雨卡莱特凯视达光电转换器
热门文章
- php子类没有构造函数父类有构造函数_如何向父类构造函数传递参数,如果父类中存在构造函数并且需要参数,子类在造对象的时候也应该传入相应的参数( )...
- win10系统扩大C盘大小
- Python 子类继承父类
- python学习小组分组程序_自动分配小组更省事点击查看_云分组小程序
- html转pdf工具安卓版,html转pdf工具(wkhtmltopdf)v2018 官方版
- Apache 源码安装详细教程
- 打开DXVA(显卡硬解码H.264与VC-1)已如此简单
- api 微信小程序组件库colorui_微信小程序常用的几个UI组件库
- 都铎王朝第一至四季/全集The Tudors迅雷下载
- 12.写一个简单的发布器与订阅器(C++)