jingsongliujing/OnnxOCR

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

Opened this issue · 2 comments

如题,需要修改哪一些内容?
直接修改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

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

收到,谢谢