Convolutional Recurrent Neural Network
December 19, 2019 ยท View on GitHub
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.pyin 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_pathin predict.py - use following script to predict
python3 predict.py