WenmuZhou / PytorchOCR

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

请问目前转换成onnx格式推理的功能可以使用了吗 #272

Closed Arlen-yuzu closed 1 year ago

Arlen-yuzu commented 1 year ago

模型从pth格式权重文件转为onnx格式后,推理结果变得不可用,在pth格式推理时还能正确预测,到了onnx格式虽然代码没有运行bug,但是输出得结果完全不可用,

Arlen-yuzu commented 1 year ago

ch_PP-OCRv4_rec这个模型在转onnx后使用会出现识别问题

WenmuZhou commented 1 year ago

可以了,你转换的时候加载权重了吗

Arlen-yuzu commented 1 year ago

可以检测了,可能是之前转换的过程有点问题,ch_PP-OCRv4_rec我重新转了一遍,现在是能够正常检测的