WenmuZhou / PytorchOCR

基于Pytorch的OCR工具库,支持常用的文字检测和识别算法

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

2018-Summer opened this issue · comments

[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 加载自己训练好的模型文件,目前没有加载模型文件,执行该推理脚本,也可以产生推理结果.

-o Global.pretrained_model=xxx.pth

(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文件在做推理,貌似也不太对。麻烦您指导一下。

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

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

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

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