jacklee1 / crnn Goto Github PK

English | 简体中文

A crnn project based on Tensorflow2.

1. Refactoring.

• Python 3.6+
• Tensorflow 2.2.0+

2020-10-13: The first version is online. 2021-3-6: Open source data construction tools 。

• Installation
  • python
  • docker
• Usage
  • train
  • test

Install the required Python packages,

pip install -r requirements.txt

1. Pull code

git clone https://github.com/lvjianjin/crnn.git

2. Prepare the data set

Generate simulation data through this tool, and modify the local dataset path to dataset_path in ./configs/config.py.

Generate simulation data through this tool or download the following Chinese data set to the local, and modify the local data set path to dataset_path in ./configs/config.py.

链接：https://pan.baidu.com/s/1FgdITVrM_HsyNh7QSpePjw 
提取码：iakr
解压密码:chineseocr

3. training

执行训练代码,

python main.py -m train

Execute training code,

python main.py -m test

The pre-trained model will be released soon, so stay tuned.

1. e-mail：[email protected]
2. QQ Group：1081332609

The codes are released under the MIT License.