OCR-transformer icon indicating copy to clipboard operation
OCR-transformer copied to clipboard

OCR seq2seq resnet+transformer

OCR-transformer

CNN + transformer model

Using colab

Preferences

Edit it hparams.py

Put the training images files in train/images/

Put the training text files in train/words/

Training

Run python ocr.py -r t

Prediction

Run python ocr.py