WenmuZhou / crnn.gluon

A gluon re-implementation of Convolutional recurrent network in gluon

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

  1. config the dataset['train']['dataset']['data_path'],dataset['validate']['dataset']['data_path'] in config.yaml
  2. generate alphabet use fellow script to generate alphabet.py in the some folder with train.py
python3 utils/get_keys.py
  1. use following script to run
python3 train.py --config_path config.yaml

Predict

predict.py is used to inference on single image

  1. config model_path, img_path in predict.py
  2. use following script to predict
python3 predict.py

About

A gluon re-implementation of Convolutional recurrent network in gluon

License:Apache License 2.0


Languages

Language:Python 100.0%