textGenerationTool

KO | EN

gitlite — search

#javascript #python #react #typescript #android #machine-learning #rust #java #windows #kubernetes #awesome-list #ios #openai #open-source

textGenerationTool

(★ 14)

ocr训练文本生成工具

textGenerationTool 최신버젼 다운로드

최종 버전 다운로드 (.zip)

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Text Renderer
Generate text images for training deep learning OCR model (e.g. [CRNN](https://github.com/bgshih/crnn)).
Support both latin and non-latin text.

# Setup
- Ubuntu 16.04
- python 3.5+

Install dependencies:
```
pip3 install -r requirements.txt
```

# Demo
By default, simply run `python3 main.py` will generate 20 text images
and a labels.txt file in `output/default/`.

![example1.jpg](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/example1.jpg)
![example2.jpg](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/example2.jpg)

![example3.jpg](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/example3.jpg)
![example4.jpg](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/example4.jpg)

# Use your own data to generate image
1. Please run `python3 main.py --help` to see all optional arguments and their meanings.
And put your own data in corresponding folder.

2. Config text effects and fraction in `configs/default.yaml` file(or create a
new config file and use it by `--config_file` option), here are some examples:

|Effect name|Image|
|------------|----|
|Origin(Font size 25)|![origin](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/origin.jpg)|
|Perspective Transform|![perspective](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/perspective_transform.jpg)|
|Random Crop|![rand_crop](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/random_crop.jpg)|
|Curve|![curve](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/curve.jpg)|
|Light border|![light border](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/light_border.jpg)|
|Dark border|![dark border](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/dark_border.jpg)|
|Random char space big|![random char space big](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/random_space_big.jpg)|
|Random char space small|![random char space small](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/random_space_small.jpg)|
|Middle line|![middle line](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/line_middle.jpg)|
|Table line|![table line](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/line_table.jpg)|
|Under line|![under line](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/line_under.jpg)|
|Emboss|![emboss](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/emboss.jpg)|
|Reverse color|![reverse color](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/reverse.jpg)|
|Blur|![blur](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/blur.jpg)|
|Text color|![font_color](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/colored.jpg)|
|Line color|![line_color](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/effects/table.jpg)|

3. Run `main.py` file.

# Strict mode
For no-latin language(e.g Chinese), it's very common that some fonts only support
limited chars. In this case, you will get bad results like these:

![bad_example1](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/bad_example1.jpg)

![bad_example2](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/bad_example2.jpg)

![bad_example3](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/bad_example3.jpg)

Select fonts that support all chars in `--chars_file` is annoying.
Run `main.py` with `--strict` option, renderer will retry get text from
corpus during generate processing until all chars are supported by a font.

# Tools
You can use `check_font.py` script to check how many chars your font not support in `--chars_file`:
```bash
python3 tools/check_font.py

checking font ./data/fonts/eng/Hack-Regular.ttf
chars not supported(4971):
['第', '朱', '广', '沪', '联', '自', '治', '县', '驼', '身', '进', '行', '纳', '税', '防', '火', '墙', '掏', '心', '内', '容', '万', '警','钟', '上', '了', '解'...]
0 fonts support all chars(5071) in ./data/chars/chn.txt:
[]
```

# Generate image using GPU
If you want to use GPU to make generate image faster, first compile opencv with CUDA.
[Compiling OpenCV with CUDA support](https://www.pyimagesearch.com/2016/07/11/compiling-opencv-with-cuda-support/)

Then build Cython part, and add `--gpu` option when run `main.py`
```
cd libs/gpu
python3 setup.py build_ext --inplace
```

# Debug mode
Run `python3 main.py --debug` will save images with extract information.
You can see how perspectiveTransform works and all bounding/rotated boxes.

![debug_demo](https://raw.githubusercontent.com/SeventhBlue/textGenerationTool/master/imgs/debug_demo.jpg)

# Todo
See https://github.com/Sanster/text_renderer/projects/1

textGenerationTool

textGenerationTool 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

textGenerationTool

textGenerationTool 최신버젼 다운로드

오늘의 추천 프로젝트

Was this content helpful?

Recent Feedback

Download README