Project Icon

tesstrain

Tesseract 5训练流程自动化工具

tesstrain是一个基于Makefile的Tesseract 5训练工作流工具。它提供完整的OCR模型训练流程,包括数据准备、训练、评估和可视化。支持自定义模型、微调和从头训练,具有灵活的配置选项。tesstrain能生成traineddata文件和错误率图表,适用于高效开发和优化Tesseract模型。

tesstrain

用于 Tesseract 5 的训练工作流程,以 Makefile 形式实现依赖跟踪。

安装

辅助工具

你至少需要 GNU make(最低版本 4.2)、wgetfindbashunzip

Leptonica, Tesseract

你需要最新版本(>= 5.3)的 Tesseract,并且需要包含训练工具和匹配的 Leptonica 绑定。 编译 说明 以及更多信息可以在 Tesseract 用户手册 中找到。

Windows

  1. 安装最新版 Tesseract(例如从 https://digi.bib.uni-mannheim.de/tesseract/ 下载),并确保 Tesseract 已添加到你的 PATH 中。
  2. 安装 Python 3
  3. 安装 Git SCM for Windows - 它提供了许多 Linux 工具在 Windows 上的版本(如 findunziprm),并将 C:\Program Files\Git\usr\bin 添加到 PATH 变量的开头(临时可以在 cmd 中使用 set PATH=C:\Program Files\Git\usr\bin;%PATH% 来实现)- 不幸的是,有几个 Windows 工具与 Linux 上的同名工具(findsort)有不同的行为/功能,需要在训练过程中避免使用它们。
  4. 安装 winget/Windows 包管理器,然后运行 winget install ezwinports.makewinget install wget 来安装缺失的工具。

Python

你需要最新版本的 Python 3.x。图像处理使用 Python 库 Pillow。 如果你没有全局安装,请使用提供的 requirements 文件 pip install -r requirements.txt

语言数据

Tesseract 需要一些配置数据(radical-stroke.txt 文件和所有脚本的 *.unicharset)放在 DATA_DIR 中。 要获取它们:

make tesseract-langdata

(虽然这一步只需要执行一次,并且隐含在 training 目标中,但你可能想要提前显式运行它。)

使用方法

选择模型名称

为你的模型选择一个名称。按照惯例,包含特定语言资源的 Tesseract 堆叠模型使用 ISO 639 中定义的(小写)三字母代码,并用下划线分隔附加信息。例如,chi_tra_vert 表示垂直排版的繁体中文。与语言无关(即特定于脚本)的模型使用大写的脚本类型名称作为标识符。例如,Hangul_vert 表示垂直排版的谚文脚本。在以下内容中,模型名称用 MODEL_NAME 表示。

提供真实数据

将由行图像和转录文本组成的真实数据放在 data/MODEL_NAME-ground-truth 文件夹中。这些文件将被分为训练和评估数据,比例由 RATIO_TRAIN 变量定义。

图像必须是 TIFF 格式并具有 .tif 扩展名,或者是 PNG 格式并具有 .png.bin.png.nrm.png 扩展名。

转录文本必须是单行纯文本,并且与行图像具有相同的名称,但图像扩展名替换为 .gt.txt

该仓库包含一个带有样本真实数据的 ZIP 存档,请参见 ocrd-testset.zip。将其解压到 ./data/foo-ground-truth 并运行 make training

注意: 如果你想从整页生成用于转录的行图像,请参见 issue 7 中的提示,特别是 @Shreeshrii 的 shell 脚本

训练

运行

make training MODEL_NAME=name-of-the-resulting-model

这是以下命令的快捷方式

make unicharset lists proto-model tesseract-langdata training MODEL_NAME=name-of-the-resulting-model

运行 make help 查看所有可用的目标和变量:

选择训练方案

首先,决定你想要进行哪种类型的训练

  • 微调:选择(并安装)一个 START_MODEL
  • 从头开始:指定一个 NET_SPEC(参见文档

更改目录假设

要覆盖默认的路径名要求,只需在上面的列表中设置相应的变量:

make training MODEL_NAME=name-of-the-resulting-model DATA_DIR=/data GROUND_TRUTH_DIR=/data/GT

如果你想使用shell变量来覆盖make变量(例如因为你正在从脚本或其他makefile运行tesstrain),那么你可以使用-e标志:

MODEL_NAME=结果模型的名称 DATA_DIR=/data GROUND_TRUTH_DIR=/data/GT make -e training

制作模型文件(traineddata)

当训练完成后,它会写入一个可以用于Tesseract文本识别的traineddata文件。请注意,这个文件不包含字典。因此tesseract可执行文件会打印一个警告。

也可以从中间训练结果(所谓的检查点)创建额外的traineddata文件。这甚至可以在训练仍在进行时完成。例如:

# 添加MODEL_NAME和OUTPUT_DIR,与训练时相同。
make traineddata

这将在OUTPUT_DIR中创建两个目录tessdata_besttessdata_fast,每个检查点都有最佳(基于双精度)和快速(基于整数)模型。

也可以只为选定的检查点创建模型。例如:

# 为最近三周的检查点文件制作traineddata。
make traineddata CHECKPOINT_FILES="$(find data/foo -name '*.checkpoint' -mtime -21)"

# 为最后两个检查点文件制作traineddata。
make traineddata CHECKPOINT_FILES="$(ls -t data/foo/checkpoints/*.checkpoint | head -2)"

# 为字符错误率(CER)优于1%的所有检查点文件制作traineddata。
make traineddata CHECKPOINT_FILES="$(ls data/foo/checkpoints/*[^1-9]0.*.checkpoint)"

如有需要,添加MODEL_NAMEOUTPUT_DIR,并将data/foo替换为输出目录。

绘制CER图

可以使用Matplotlib绘制训练和评估字符错误率(CER):

# 生成OUTPUT_DIR/MODEL_FILE.plot_*.png
make plot

上面定义的所有变量都适用,但没有对training的显式依赖。

尽管如此,该目标仍然依赖于训练期间捕获的LOG_FILE(只是不会触发训练本身)。除了分析日志文件外,这还直接在评估数据集上评估训练好的模型(对每个检查点)。后者也可作为独立目标evaluation使用:

# 生成OUTPUT_DIR/eval/MODEL_FILE*.*.log
make evaluation

即使在训练仍在进行时也可以进行绘图,并将描绘到那时为止的训练状态。(只要LOG_FILE发生变化或写入新的检查点,就可以随时重新运行。)

作为示例,使用ocrd-testset.zip中提供的训练数据进行一些训练并生成图表:

unzip ocrd-testset.zip -d data/ocrd-ground-truth
make training MODEL_NAME=ocrd START_MODEL=frk TESSDATA=~/tessdata_best MAX_ITERATIONS=10000 &
# 生成data/ocrd/ocrd.plot_cer.png和plot_log.png(在训练期间/之后重复)
make plot MODEL_NAME=ocrd

然后应该看起来像这样:

ocrd.plot_cer.png

许可证

软件根据Apache 2.0许可证的条款提供。

Deutsches Textarchiv提供的样本训练数据属于公共领域

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号