Mask TextSpotter v3 来了!最强端到端文本识别模型

Stella981
• 阅读 1093

Mask TextSpotter v3 来了!最强端到端文本识别模型

场景文本的识别可以用文本检测+文本识别两个过程来做,近年来端到端的场景文本识别(即Text Spotting)越来越引起学术界的重视,而华中科技大学白翔老师组的 Mask TextSpotter v1、v2 一直是该领域的代表性工作。

近日  Mask TextSpotter v3 发布,代码已开源,论文 Mask TextSpotter v3: Segmentation Proposal Network for Robust Scene Text Spotting 详细介绍了其要解决的问题、使用的方案和达到的性能,此次更新依然带来惊喜,在多个数据集上大幅刷新了SOTA!

该文作者信息:

Mask TextSpotter v3 来了!最强端到端文本识别模型

作者来自华中科技大学和 Facebook AI。

文中称 Mask TextSpotter v3  在 ICDAR 2013 数据集上提升了 21.9个百分点,在Total-Text 数据集上提升了5.9个百分点,在MSRA-TD500 数据集上也达到了SOTA,证明该算法具有强大的应对文本旋转、形状多变和极端纵横比的鲁棒性。

要解决的问题

如下图:

Mask TextSpotter v3 来了!最强端到端文本识别模型

之前的端到端文本识别模型使用包围文本的矩形框作为Proposals,在应对旋转、形状任意和极端纵横比的文本时存在明显的缺陷。图中(a)内部的Proposals对应的特征包含邻接文本的特征,不能很好的识别文本。

而Mask TextSpotter v3使用一种称之为分割候选网络( Segmentation Proposal Network )生成多边形Proposals,并在此Proposals基础上计算hard ROI 特征,更好的表征了文本区域,可以取得更好的识别结果。

Mask TextSpotter v3 的关键是如何得到包围文本的多边形Proposals(通过分割的思想),和硬加权的hard ROI 特征,作者的实验证明去除干扰的hard ROI 特征可大幅提高识别精度。

Mask TextSpotter v3整体流程

如下图:

Mask TextSpotter v3 来了!最强端到端文本识别模型

Mask TextSpotter v3 使用U-Net结构提取分割的特征,同FPN不同,v3 直接将不同的尺度特征融合成特征 F,使用 F 做文本区域的预测,对预测得到的在(0,1)之间的连续值结果,使用DB(Real-time scene text detection with differentiable binarization,AAAI 2020)二值化得到多个文本区域,然后将连通的区域作为一个文本实例的收缩(这里是关键!),对其进行一定系数的膨胀操作,即得到了文本Proposal。

得到文本区域 Proposal 后,仅含有0和1元素的二值多边形mask矩阵与特征按元素相乘,得到Hard ROI masking 特征。此处名中的Hard(硬)因为这里是仅含有0与1的二值加权,ROI masking 操作按照多边形 mask 将特征在表示对应区域上“提精”了。

到这里既找到了文本区域,又有了每个文本区域对应的特征,作者按照Mask TextSpotter v2 的流程进行识别,使用Fast R-CNN算法,字符分割模块和空间注意力模型都被用于识别。

在训练的时候,要制作分割标签,下图为标签生成示意图:

Mask TextSpotter v3 来了!最强端到端文本识别模型

Mask TextSpotter v3 针对  v2 中的 Proposal 的产生进行了改进,并利用Hard ROI masking 提精了文本实例对应的ROI特征,可以在文本检测的三大难题(旋转、形状任意、极端纵横比)中得到更好的结果。

实验结果

作者在多个数据集上进行了实验,以下为在三个数据集上识别结果可视化:

应对旋转,比v2更好:

Mask TextSpotter v3 来了!最强端到端文本识别模型

应对极端纵横比,比v2更好:

Mask TextSpotter v3 来了!最强端到端文本识别模型

应对形状任意文本,比v2更好:

Mask TextSpotter v3 来了!最强端到端文本识别模型

以下为在ROIC13数据集上在检测任务和端到端识别任务中的结果比较,大幅超越之前的SOTA:

Mask TextSpotter v3 来了!最强端到端文本识别模型

以下为在MSRA-TD500数据集上在检测任务中的结果:

Mask TextSpotter v3 来了!最强端到端文本识别模型

在Total-Text 数据集上端到端识别结果:

Mask TextSpotter v3 来了!最强端到端文本识别模型

应对小实例文本数据集,在IC15上的结果比较,v3 取得了三个最好结果:

Mask TextSpotter v3 来了!最强端到端文本识别模型

总结与思考

Mask TextSpotter v3 在端到端文本识别任务中实现了更高的精度,得益于从分割中获得文本多边形区域表示和Hard ROI masking 特征提精方法。

值得一提的是, v3 基于 v2 和可微分二值化方法 DB 等,这些研究都出自华科白翔老师组,体现出作者们对问题的"系统性"思考而产生的不断的技术创新。

论文地址:

https://arxiv.org/abs/2007.09482

代码地址:

https://github.com/MhLiao/MaskTextSpotterV3

在我爱计算机视觉公众号后台回复“MaskTS”,即可收到论文和代码下载。

Mask TextSpotter v3 来了!最强端到端文本识别模型

备注:ocr

Mask TextSpotter v3 来了!最强端到端文本识别模型

OCR交流群

文本检测、识别、编辑等更多最新技术,若已为CV君其他账号好友请直接私信。

我爱计算机视觉

微信号:aicvml

QQ群:805388940

微博知乎:@我爱计算机视觉

投稿:amos@52cv.net

网站:www.52cv.net

Mask TextSpotter v3 来了!最强端到端文本识别模型

在看,让更多人看到 Mask TextSpotter v3 来了!最强端到端文本识别模型

本文分享自微信公众号 - 我爱计算机视觉(aicvml)。
如有侵权,请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”,欢迎正在阅读的你也加入,一起分享。

点赞
收藏
评论区
推荐文章
blmius blmius
3年前
MySQL:[Err] 1292 - Incorrect datetime value: ‘0000-00-00 00:00:00‘ for column ‘CREATE_TIME‘ at row 1
文章目录问题用navicat导入数据时,报错:原因这是因为当前的MySQL不支持datetime为0的情况。解决修改sql\mode:sql\mode:SQLMode定义了MySQL应支持的SQL语法、数据校验等,这样可以更容易地在不同的环境中使用MySQL。全局s
待兔 待兔
4个月前
手写Java HashMap源码
HashMap的使用教程HashMap的使用教程HashMap的使用教程HashMap的使用教程HashMap的使用教程22
不是海碗 不是海碗
1年前
一文带你看透通用文字识别 OCR
随着人工智能技术的不断发展,通用文字识别OCR也变得越来越重要。通用文字识别OCR是指对文本图像进行分析,以转换图像中的文本为可用的文本形式的过程,其主要用途是提供文本可搜索和复制的功能。
Stella981 Stella981
3年前
Android So动态加载 优雅实现与原理分析
背景:漫品Android客户端集成适配转换功能(基于目标识别(So库35M)和人脸识别库(5M)),导致apk体积50M左右,为优化客户端体验,决定实现So文件动态加载.!(https://oscimg.oschina.net/oscnet/00d1ff90e4b34869664fef59e3ec3fdd20b.png)点击上方“蓝字”关注我
Stella981 Stella981
3年前
Google研究人员推出了一种用于生成文本到图像的新框架(TReCS)
!(https://oscimg.oschina.net/oscnet/faedcb264a1c43969f2f5a2e6b9dbd2e.png)基于生成对抗网络(GAN)的深度神经网络促进了端到端可训练的照片级逼真的文本到图像的生成。许多方法还使用中间场景图表示法来改善图像合成。使用基于对话的交互的方法允许用户提供指令,以逐步改进和调整生成
卷积神经网络表征可视化研究综述
卷积神经网络表征可视化研究综述(1)转载自:人工智能技术与咨询源自:自动化学报作者:司念文张文林屈丹罗向阳常禾雨牛铜摘要近年来,深度学习在图像分类、目标检测及场景识别等任务上取得了突破性进展,这些任务多以卷积神经网络为基础搭建识别模型,训练后的模型拥有优异的自动特征提取和预测性能,能够为用户提供“输入–输出”形式的端到端解决方案.然而,
流浪剑客 流浪剑客
1年前
Mac端文本识别工具:TextSniper for Mac激活版 支持M1
是一款易于使用的桌面OCR应用程序,它可以帮助用户快速提取和识别Mac屏幕上不可搜索和不可编辑的文本。这款工具的使用非常直观,不论是图像、扫描的纸质文档、PDF还是视频中的文本,都可以轻松地从其中提取文本。作为一项额外功能,TextSniper可以将OCR
铁扇公主 铁扇公主
1年前
Mac哪款文本识别工具好用呢?TextSniper mac版介绍
Mac哪款文本识别工具好用呢?TextSnipermac版是一款运行在MacOS平台上好用的OCR文本识别工具。TextSniper可以快速捕捉任何文本,包括演示文稿,培训,屏幕广播,图像,图片,网页,视频教程,照片,电子书,PDF等抓取和识别文本。操作简
四儿 四儿
1年前
语音识别技术:端到端的挑战与解决方案
一、引言随着人工智能技术的不断发展,语音识别技术得到了越来越广泛的应用。端到端语音识别技术是近年来备受关注的一种新型语音识别技术,它能够直接将语音转换成文本,省略了传统的语音特征提取步骤。本文将探讨端到端语音识别技术的挑战与解决方案。二、端到端语音识别技术
崇恩圣帝 崇恩圣帝
6个月前
使用机器学习识别文本验证码的实现方法
在网络应用中,验证码常用于防止恶意机器人或自动程序对网站进行攻击。识别文本验证码是一个常见的问题,本文将介绍如何使用机器学习方法来实现这一目标。1.文本验证码识别方法文本验证码通常包含英文和数字字符,识别文本验证码的一种常见方法是使用图像处理和机器学习技术