4 个月前

金字塔掩模文本检测器

金字塔掩模文本检测器

摘要

场景文字检测是场景文字识别系统中的一个关键步骤,旨在自动定位自然场景图像中的文字实例。近期的一些研究得益于Mask R-CNN,将场景文字检测任务视为实例分割问题,并取得了显著的性能提升。在本文中,我们提出了一种基于Mask R-CNN的新框架,称为金字塔掩模文字检测器(Pyramid Mask Text Detector, PMTD),用于处理场景文字检测。与现有的基于Mask R-CNN的方法生成的二值文本掩模不同,我们的PMTD在位置感知监督的指导下执行像素级回归,为每个文本实例生成更具信息量的软文本掩模。对于文本框的生成,PMTD将获得的二维软掩模重新解释为三维空间,并引入了一种新颖的平面聚类算法,以基于三维形状推导出最优的文本框。标准数据集上的实验表明,所提出的PMTD带来了持续且显著的性能提升,并明显优于现有最先进方法。具体而言,在ICDAR 2017 MLT数据集上,PMTD实现了80.13%的F值。

代码仓库

基准测试

基准方法指标
scene-text-detection-on-icdar-2015PMTD
F-Measure: 89.33
Precision: 91.3
Recall: 87.43
scene-text-detection-on-icdar-2017-mlt-1PMTD*
F-Measure: 80.13%
Precision: 84.42
Recall: 76.25

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
金字塔掩模文本检测器 | 论文 | HyperAI超神经