search

WenmuZhou / PytorchOCR

基于Pytorch的OCR工具库，支持常用的文字检测和识别算法
1.39k stars 307 forks source link

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

Closed 2018-Summer closed 1 year ago

2018-Summer commented 1 year ago

[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 commented 1 year ago

-o Global.pretrained_model=xxx.pth

2018-Summer commented 1 year ago

(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 commented 1 year ago

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

2018-Summer commented 1 year ago

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

WenmuZhou commented 1 year ago

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

2018-Summer commented 1 year ago

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