Use trOCR in onnx format
#3
by
fef2
- opened
Hi, I have trained the small-printed trocr on my custom dataset having multiline images like you and converted it in onnx. The trained model can read full text. But while converting the model to onnx, the model detects only first line or part of it in first line. I have used this https://github.com/huggingface/transformers/issues/19811#issuecomment-1303072202
for inference and this command "optimum-cli export onnx -m {model_checkpoints} --task vision2seq-lm onnx/ --atol 1e-3" for convert to onnx
Have you ever encountered this problem?