/TextGenerator

OCR dataset Text-Detection dataset Font-Classification dataset generator

Primary LanguagePython

TextGenerator

  • This is a tools for ocr dataset, text detection, fonts classification dataset generate.
  • 这是一个用来生成ocr数据,文字检测数据,字体识别的最方便的工具

实现的功能:

  • 生成基于不同语料的,不同字体、字号、颜色、旋转角度的文字贴图
  • 支持多进程快速生成
  • 文字贴图按照指定的布局模式填充到布局块中
  • 在图像中寻找平滑区域当作布局块
  • 支持文字区域的图块抠取导出(导出json文件,txt文件和图片文件,可生成voc数据,ICDAR_LSVT数据集格式!)
  • 支持每个文字级别的标注(存入了lsvt的json文件中)
  • 支持用户自己配置各项生成配(图像读取,生成路径,各种概率)

效果预览

生成图片示例:

文字贴图示例:

旋转矩形示例

单个文字包围框示例

使用方式

  • 环境安装(Python3.6+,建议使用conda环境)

    # step 1
    pip install requirements.txt
    # step 2
    sh make.sh
    
  • 编辑配置文件config.yml(可选)

  • 执行生成脚本

    python3 run.py
    
  • 生成的数据

    生成的数据存放在config.yml中的provider> layout> out_put_dir指定的目录下。