Awesome
Convolutional Recurrent Neural Network
This software implements the Convolutional Recurrent Neural Network (CRNN) in gluon. Origin software could be found in crnn
Requirements
- mxnet 1.5.0
- mxboard
Data Preparation
Prepare a text in the following format
/path/to/img/img.jpg label
...
Train
- config the
dataset['train']['dataset']['data_path']
,dataset['validate']['dataset']['data_path']
in config.yaml - generate alphabet
use fellow script to generate
alphabet.py
in the some folder withtrain.py
python3 utils/get_keys.py
- use following script to run
python3 train.py --config_path config.yaml
Predict
predict.py is used to inference on single image
- config
model_path
,img_path
in predict.py - use following script to predict
python3 predict.py