Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

python tools/infer_rec.py -c configs/rec/PP-OCRv3/ch_PP-OCRv3_rec_distillation.yml #271

Closed
2018-Summer opened this issue Sep 1, 2023 · 6 comments

Comments

@2018-Summer
Copy link

[2023/09/01 11:34:19] torchocr INFO: infer_img: C:/Users/94752/PycharmProjects/TransformerMeter/PytorchOCR-dev/doc/imgs_words/en/192_crop_0.jpg
[2023/09/01 11:34:19] torchocr INFO: C:/Users/94752/PycharmProjects/TransformerMeter/PytorchOCR-dev/doc/imgs_words/en/192_crop_0.jpg result: 9 0.08595956861972809
[2023/09/01 11:34:19] torchocr INFO: success!

大佬怎么在 infer_rec.py 加载自己训练好的模型文件,目前没有加载模型文件,执行该推理脚本,也可以产生推理结果.

@WenmuZhou
Copy link
Owner

-o Global.pretrained_model=xxx.pth

@2018-Summer
Copy link
Author

(venv) C:\Users\94752\PycharmProjects\TransformerMeter\PytorchOCR-dev>python tools/infer/predict_rec.py --rec_model_dir=output/rec/rec_ppocr_v3/export/ --image_dir=dataOCR/DBXdata/image/192_crop_0.jpg
[2023/09/04 16:53:11] torchocr INFO: result of dataOCR/DBXdata/image/192_crop_0.jpg:('1', 0.08545342087745667)

大佬 ,再请教一下,导出onnx文件在做推理,貌似也不太对。麻烦您指导一下。

@WenmuZhou
Copy link
Owner

导出之前的模型是训练好的模型吗

@2018-Summer
Copy link
Author

嗯嗯是的,-o Global.pretrained_model=xxx.pth ,把您刚刚说这个参数加上,推理结果也是正确的,但是导出onnx在推理就出现了上面得问题

@WenmuZhou
Copy link
Owner

导出的时候加上上面那句话了吗

@2018-Summer
Copy link
Author

可以了,重新运行了一下导出onnx文件,推理结果正确了,非常感谢.....

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants