本范例将使用OCR技术实现蒙古文书籍文字识别检测和识别,通过本章您可以掌握:
本范例由内蒙古阿尔泰电子信息技术有限公司CTO欧日乐克、苏日图、达布希腊图、歆琪乐与飞桨联合打造
蒙古文文字识别技术在蒙古文信息处理领域成为一个亟待解决的问题。但由于诸多主客观原因,产品化蒙古文文字识别技术上有一段距离。而书籍的数字化是其中一项重要的课题,但因为涉及相关技术较多,难度较高,在整体上解决蒙古文书籍文字识别技术还不成熟。
*基本概念
字:即字符,不同于汉字的字(Character)一字一形,并包含字意,蒙古文的字类似于英语的
字母(Letter)是一种表音文字,但却拥有若干字形(Glyph),若干个字构成一个词。
名义字符:Unicode十大法则中规定,"Character is not Glyph"(字符不是字形)。
于是像蒙古文一样有形态变化的文字就有了一个字形代表其他形态的字形。该字形被记录在
Unicode基本表中并分配编码。如蒙古文 ᠠ([a]) 为 a 的独立形式,分配码位\u1820,
记作uni1820.isol,代表所有词首uni1820.init、词中uni1820.medi和词尾1820.fina形式。
变形显现字符:除名义字符外的其他没有实际编码位置的字形。在蒙古文文本中看到的字符皆为通过
Unicode Scripts Processor(简称USP)处理后显示出来的字符。
词:语言里最小的可以独立运用的单位,等同于英语的word,以空格分割的字符串。
一形多字
多字一形
书写方向与换行方向会影响识别后的排序以及文字方向。
蒙古文中有一些有关形态变化的控制字符,其中最频繁出现的有 Mongolian Vowel Separator (MVS,\u180E),和 Narrow No-Break Space(NNBSP,\u202F)。该两个控制符正常 情况下在文本中是透明的,比空格窄,不能换行。MVS用于连接词干与词尾a、e元音,NNBSP则 用于连接词与词附加成分(可以理解成分写后缀)。MVS会引发双向形变,而NNBSP则会引发后位 形变。
此外,蒙古文中有一些字形为另一个字形的组成部件,导致识别时不好对应结果。
针对以上问题, 本例选用PP-OCRv3这一开源超轻量OCR系统进行蒙古文文本识别系统的开发。我们首先使用数据合成工具合成了250万数据,基于这部分数据进行训练,通过精准切除白色边缘,随机加入标点符号,调整评估集数据使识别精度提升至75.78%。效果如下:
策略 | 精度 % |
---|---|
合成数据训练 | 42.38 |
优化合成训练 | 75.78 |
具体流程为:
评估集数据的质量对模型的训练效率也起到很大的作用。
本任务基于Aistudio完成, 具体环境如下:
下载PaddlleOCR代码并安装依赖库:
git clone -b dygraph https://gitee.com/paddlepaddle/PaddleOCR
# 安装依赖库
cd PaddleOCR
pip install -r PaddleOCR/requirements.txt
将下载或训练完成的模型放置在PaddleOCR/output
下, 然后使用如下命令即可快速进行模型推理
python tools/infer_rec.py -c configs/rec/PP-OCRv3/multi_language/Mongolian_PP-OCRv3_rec.yml \
-o Global.pretrained_model=output/v3_Mongolian_mobile/best_accuracy \
Global.infer_img=doc/imgs_words/
本项目从真实语料中生成250万张图片作为训练集。另外生成1万张图片作为验证集。
蒙古文由于编码原因生成图片时不能直接用 Unicode 字符串生成。蒙古文 Unicode 的本质是音码,伴随复杂的形态变化,如果没有对应的复杂文本处理程序则只能显示蒙古文的名义字符,而非变形显现字符。 因此如果想生成蒙古文图片则需要: 1.调用Windows系统的 USP10.dll,2.用支持形码的字体生成。 本项目使用了第二种方案,即使用形码的字体生成图片,并对应 Unicode 标签。
直接使用 Unicode 生成的情况(字符会分开并以名义字符显示):
$$\mbox{左列为Unicode生成图片,右列为Unicode文本}$$
$$\mbox{左列为Unicode文本,右列为形码生成图片}$$
生成图片时建议将字符串长度保持在5个词(平均30个字符),否则训练较为困难。
部分训练图片示例如下:
为验证模型对实际图片的效果,验证图片采用了真实扫描图片。在扫描完整的页面后对
标签文件格式如下:
数据集类型 | 数量 |
---|---|
训练集 | 250万 |
验证集 | 1.1万 |
数据文件结构如下:
PaddleOCRv3
├── train_data # 训练数据文件夹
│ ├── texts
│ │ ├── train1.txt # 生成的训练数据标签,与图片文档一一对应
│ │ ├── train2.txt
│ │ ├── train3.txt
│ │ ├── train4.txt
│ │ ├── train11.txt
│ │ ├── train20.txt
│ │ ├── train21.txt
│ │ └── train22.txt
│ ├── image1 # 生成的训练图片
│ ├── image2
│ ├── image3
│ ├── image4
│ ├── image11
│ ├── image20
│ ├── image21
│ └── image22
├── test_data # 验证数据文件夹
│ ├── test_data.txt # 验证数据标签
│ ├── 0 # 每个文件夹有34张图片
│ ├── 1
: :
: :
│ └── 409
根据 Unicode 编码顺序制作一个包含所有蒙古文字符的文本字典,建议保存到./ppocr/utils/dict目录下面,并在yml文件中更改地址。
准备好合成数据后,我们可以使用以下命令训练数据:
cd ${PaddleOCR_root}
python tools/train.py -c configs/rec/PP-OCRv3/multi_language/Mongolian_PP-OCRv3_rec.yml
如果想从断点继续训练:
cd ${PaddleOCR_root}
python tools/train.py -c configs/rec/PP-OCRv3/multi_language/Mongolian_PP-OCRv3_rec.yml \
-o Global.checkpoints=./output/v3_Mongolian_mobile/best_accuracy
可填各参数含义如下:
-c: 指定使用的配置文件,Mongolian_PP-OCRv3_rec.yml对应于OCRv3识别模型。
-o: 覆盖配置文件中参数
Global.pretrained_model: 指定使用的预训练模型
Global.checkpoints: 断点位置
Global.epoch_num: 指定训练的epoch数
Global.eval_batch_step: 间隔多少step做一次评估
Train.dataset.data_dir: 训练数据集路径
Train.dataset.label_file_list: 训练集文件列表
Train.loader.batch_size_per_card: 训练单卡batch size
Eval.dataset.data_dir: 评估数据集路径
Eval.dataset.label_file_list: 评估数据集文件列表
Eval.loader.batch_size_per_card: 评估单卡batch size
训练好的模型推测如下:
python tools/infer_rec.py -c configs/rec/PP-OCRv3/multi_language/Mongolian_PP-OCRv3_rec.yml \
-o Global.pretrained_model=output/v3_Mongolian_mobile/best_accuracy \
Global.infer_img=doc/imgs_words/
训练完成后可以测试模型。可将测试图片指定到某文件夹:
PaddleOCRv3
├── doc
├── imgs_words
│ ├── arabic
│ ├── belarusian
│ ├── bulgarian
: :
: :
│ ├── mongolian # 在此放入真实蒙古文图片,一个图片一行
│ └── uyghur
快速推测
python tools/eval.py -c configs/rec/PP-OCRv3/multi_language/Mongolian_PP-OCRv3_rec.yml \
-o Global.checkpoints=./output/v3_Mongolian_mobile/best_accuracy
推测结果将被记录在predicts_ppocrv3_Mongolian.txt文件中。
PaddleOCRv3
├── output
│ ├── rec
│ │ └── predicts_ppocrv3_Mongolian.txt
│ └── v3_Mongolian_mobile
部分结果:三列分别为推测结果、真实标签、图片
本例选用PP-OCRv3这一开源超轻量OCR系统进行蒙古文文本识别系统的开发。加入250万合成数据,在现有模型基础上进行微调,通过修正训练集,设定评估标准,最终将蒙古文识别精度从42%提升至75%。