rsommerfeld / trocr

Powerful handwritten text recognition. A simple-to-use, unofficial implementation of the paper "TrOCR: Transformer-based Optical Character Recognition with Pre-trained Models".
MIT License
181 stars 27 forks source link

Pretrained model #6

Closed 123akopf321 closed 2 years ago

123akopf321 commented 2 years ago

Can you also provide a pre-trained model for testing?

rsommerfeld commented 2 years ago

There is already a pre-trained model available for testing from huggingface. No need to download it manually, you can just use it by adding the --no-local-model flag. See the section on Inference (Prediction) in the readme:

python -m src predict data/* --no-local-model  # uses the pretrained huggingface model

Is this what you were thinking of?