Code for training LSTM neural network on top of convolutional features for captcha recognition in Moscow subway
All code tested with python 2.7, TF 1.0
- To download sythetically generated captchas run
python downloader.py
Images will be stored in "data/train" - Run to_tfrecords.py to create *.tfrecords files from images
- Run ocr_train.py to train network. In different terminal run
export CUDA_VISIBLE_DEVICES=""
python ocr_eval.py
- After training edit chekpoint path in
freeze.sh
with your best checkpoint path and then run it - ???
- PROFIT!
Sample of captcha image: