jingsongliujing / OnnxOCR

基于PaddleOCR重构,并且脱离PaddlePaddle深度学习训练框架的轻量级OCR,推理速度超快
Apache License 2.0
731 stars 72 forks source link

怎么用ch_ppocr_server_v2.0的模型进行推理? #13

Open eyu11 opened 4 months ago

eyu11 commented 4 months ago

如题,需要修改哪一些内容? 直接修改utils的初始化参数--rec_model_dir,会报如下的错: onnxruntime.capi.onnxruntime_pybind11_state.InvalidArgument: [ONNXRuntimeError] : 2 : INVALID_ARGUMENT : Got invalid dimensions for input: x for the following indices index: 2 Got: 48 Expected: 32 Please fix either the inputs or the model.

用V4版本会出错漏行的情况,如下图所示:

image
jingsongliujing commented 4 months ago

v2模型请切换分支到master,master分支就是v2版本的推理,实际上就是修改一下输入模型的图片大小参数 image

eyu11 commented 4 months ago

收到,谢谢