OCR数据集生成项目TextRecognitionDataGenerator

合集 - OCR问题(3)

1.OCR数据集生成项目TextRecognitionDataGenerator2024-12-17

2.CTC Loss2024-10-073.OCR数据集2024-10-07

收起

1、开源OCR数据集生成项目TextRecognitionDataGenerator

该项目通过 Python实现，可以通过 pip 安装：

终端： pip install trdg

进入文件夹,安装requirementspip install -r requirements.txt
安装完成
进入文件夹cd trdg
尝试运行看是否存在错误python run.py -c 10 -w 5
就可以生成如下图片，其中 -c 参数表示生成图片的数量， -w 表示图片中单词的个数。

2、生成数据方法

使用方法
trgd文件夹中，更换背景文件夹-images 内放入想要的背景
dicts里面放入想要的字典
out为默认输出文件夹
fonts为字体文件夹，只能识别ttf类型，需要把ttf放在对应的文件夹中
运行命令参照下列表格
修改名称：run.py中倒数几行iname 可以更改生成的txt中的文件名，从0开始填写i+0
图片生成后，使用rename.py更改名称，变量k后面的参数，一定要与run.py中的一致

参考：
https://blog.csdn.net/u012995500/article/details/109405270
https://blog.csdn.net/yuanjiaqi_k/article/details/126636864
https://www.cnblogs.com/nakkk/p/14337605.html
https://blog.csdn.net/jhsignal/article/details/107930105
https://www.cnblogs.com/Allen-rg/p/9774080.html