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

issue on shape of underlying tensors #14

Open Olivier-aka-Raiden opened 1 year ago

Olivier-aka-Raiden commented 1 year ago

Hi, I wanted to give it a try and when trying to do inferance on the base model, I get an error as follows : IndexError: The shape of the mask [1, 4] at index 0 does not match the shape of the indexed tensor [8, 5] at index 0

I tried with jpg and png image, both have the exact output. Also, using the python example code didn't change anything, error is the same. Is it because of new tensorflow version ? Can you help me fix the code to make it work ? Probably the definition of the dtype is incorrect.

rsommerfeld commented 1 year ago

Hi, have you been able to resolve your problem? If not, can you provide more details? All parts of the repository are still working on a clean setup for me. The repository also relies on pytorch, not tensorflow.