当前位置:   article > 正文

软件工程应用与实践(二):Paddle OCR的框架与技术初识

paddle ocr

2021SC@SDUSC

目录

一、Paddle OCR的介绍

1.1 OCR的发展与面临的困难 

1.2 PaddleOCR的推出 

二、PP-OCR的框架

2.1 文字检测器

2.2 方向分类器

2.3 文字识别器

三、PP-OCR各模型的策略具体实现

3.1 文本检测器模型

3.2 方向分类器模型

3.3 文字识别器模型 (主要介绍本人负责部分)


一、Paddle OCR的介绍

1.1 OCR的发展与面临的困难 

       OCR (Optical Character Recognition) 是一种以自动识别图像中的文本为目标的技术,其研究历史悠久,应用范围广泛,如文件电子化、身份认证、数字金融系统、车牌识别等。此外,在工厂中,通过自动提取产品的文本信息,可以更方便地管理产品。学生的线下作业或试卷可以通OCR 系统电子化,使师生之间的交流更加高效。OCR 还可以用于标记街景图像的兴趣点 (POI),提高 地图制作效率。丰富的应用场景赋予了 OCR 技术巨大的 商业价值,同时也带来了很多挑战。
      图像中的文本大致可分为两类:场景文本和文档文本 场景文本是指如图所示的自然场景中的文本,它通常会因为一 些因素而发生巨大的变化,如视角缩放、弯曲、混乱、字体、多语言、模糊、光照等。文档文本在实际应用中更常见。它也有高密度、长文本等问题需要解决。同时,文档图像文本识别往往需要 对结果进行结构化,这就带来了一个新的困难任务。
      在实际应用中,需要处理的图像通常是大量的,这使得高计算效率成为设计 OCR 系统的重要标准。 优先选择 CPU 而不是 GPU。
    在成本方面,特别是OCR 系统需要在许多场景下的嵌入式设备上运行,比如手机,这就需要考虑模型的大小,而权衡模型大小和性能是困难的。

1.2 PaddleOCR的推出 

      Paddle Paddle提出一种实用的超轻量级 OCR 系统,命名为 PP- OCR,该系统由文本检测、检测框校正和文本识别三部分组成。
       PP-OCR算法被开发者广泛应用,短短半年时间, 累计Star数量已超过15k ,频频登上Github Trending和Paperswithcode 日榜月榜第一, 称它为 OCR方向目前最火的repo 绝对不为过。
       其具有多方面的优势:
  • 在 PaddleOCR 识别中,会依次完成三种任务:检测、方向分类及文本识别;
  • 关于预训练权重,PaddleOCR 官网根据提供权重文件大小分为两类:轻量型和服务器部署;
  • 支持多语言识别,目前能够支持 80 多种语言;
  • 提供有丰富的 OCR 领域相关工具供我们使用,方便我们制作自己的数据集、用于训练;





二、PP-OCR的框架

2.1 文字检测器

      文本检测的目的是定位图像中的文本区域
       PP- OCR 中,我们使用可微分二值化 (DB) (Liao et al. 2020) 作为基于简单分割网络的文本检测器。DB 的简单后处理使得它非常高效。为了进一步提高其有效性和效率,采用了以下六种策略: 轻主干、轻头部、去除 SE 模块、余弦学习率衰 减、学习率预热和 FPGM 剪枝。最后,将文本检测器的模型尺寸减小到 1.4M

2.2 方向分类器

      检测框修复识别检测到的文本前,需要将文本框转换为水平矩形框进行后续的文本识别。
      由于检测帧由四个点组成,因此易于通过几何变换实现。 然而,矫正后的盒子可能会被反转。因此,需要一个分类器来确定文本的方向。如果确定一个方框是反向的,则需要进一步翻转。训练文本方向分类器是一项简单的图像分 类任务。我们采用以下四种策略来增强模型能力和缩小模型尺寸: 轻骨干网、数据增强、输入分辨率和 PACT 量化。最后,文本方向分类器模型大小为 500KB。

2.3 文字识别器

       在 PP-OCR 中,我们使用 CRNN (Shi, Bai, and Yao 2016) 作 为文本识别器,它在文本识别中被广泛应用和实用。
       CRNN融合了特征提取和序列建模。它采用了连接时态分类(CTC)的损失,避免了预测与标注不一致的问题。为了增强文本识别器的模型能力和缩小模型大小,采用了以下 9 种策略 :轻主干、数据增强、余弦学习速率衰减、特征图解析、 正则化参数、学习率预热、轻头部、预训练模型和 PACT 量化。文本识别器中、英文识别器模型尺寸仅为 1.6M ,字母、数字、符号识别器模型尺寸仅为 900KB





三、PP-OCR各模型的策略具体实现





3.1 文本检测器模型

       介绍增强文本检测器的模型能力或减少模型大小的 7 种策略。
  • 轻骨架,选用采用 MobileNetV3 large x0.5 来权衡精度和效率;
  • 轻头部,FPN (Lin et al.2017)架构;
  • 去SE,去除 SE 块后,模型尺寸从 4.1M 减小到 2.5M,但精度不受影响;
  • 余弦学习率衰减,学习率学习速度的超参数,学习率越低,损失值变化越慢;
  • 学习率预热,有助于提高图像分类的准确率;
  • FPGM,使用 FPGM (He etal. 2019b)在原始模型中寻找不重要的子网络;





3.2 方向分类器模型

       介绍增强方向分类器的模型能力或减少模型大小的 7 种策略。

  • 轻骨架,选用采用 MobileNetV3 large x0.5 来权衡精度和效率;
  • 数据增强,加 入 了 BDA 和RandAugment;
  • 加入分辨率,当一幅归一化图像的输入分辨率增加时,精度也会提高;
  • PACT量化,使神经网络模型具有更低的延迟、更小的体积和更低的计算功耗;

3.3 文字识别器模型 (主要介绍本人负责部分)

       介绍增强文字识别器的模型能力或减少模型大小的 9 种策略。

  • 轻主干,选用采用 MobileNetV3 large x0.5 来权衡精度和效率;
  • 数据增强,BDA (Base Dataaugmented)和TIA (Luo et al. 2020);
  • 余弦学习率衰减,有效提高模型的文本识别能力;
  • 特征图辨析,适应多语言识别,进行向下采样 feature map的步幅修改;
  • 正则化参数,权值衰减避免过拟合;
  • 学习率预热,同样有效;
  • 轻头部,采用全连接层将序列特征编码为预测字符,减小模型大小;
  • 预训练模型,是在 ImageNet 这样的大数据集上训练的,可以达到更快的收敛和更好的精度;
  • PACT量化,略过 LSTM 层;

学习率模型:


TLA算法 





 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/364172
推荐阅读
相关标签
  

闽ICP备14008679号