Skip to content
This repository has been archived by the owner on Oct 28, 2020. It is now read-only.

Code for training LSTM neural network on top of convolutional features for captcha recognition in Moscow subway

License

Notifications You must be signed in to change notification settings

mosmetro-android/tf-cnn-lstm-ocr-captcha

 
 

Repository files navigation

tf-cnn-lstm-ocr-captcha

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

  1. To download sythetically generated captchas run python downloader.py Images will be stored in "data/train"
  2. Run to_tfrecords.py to create *.tfrecords files from images
  3. Run ocr_train.py to train network. In different terminal run
export CUDA_VISIBLE_DEVICES=""
python ocr_eval.py
  1. After training edit chekpoint path in freeze.sh with your best checkpoint path and then run it
  2. ???
  3. PROFIT!

Sample of captcha image:

sample

About

Code for training LSTM neural network on top of convolutional features for captcha recognition in Moscow subway

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.8%
  • Shell 0.2%