search

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

score/certainty of prediction #7

Closed krispydockx closed 2 years ago

krispydockx commented 2 years ago

Hi

I was wondering if there is any way to score a prediction, how certain is the model that the output is correct.

Thanks

rsommerfeld commented 2 years ago

Take a look at the newest commit - I added confidence scores to the predict method!