PaddleOCR2Pytorch icon indicating copy to clipboard operation
PaddleOCR2Pytorch copied to clipboard
verified publisher icon frotms

请教一下，pytorch ocr crnn的resnet34 pth文件怎么转onnx？

Open nissansz opened this issue 3 years ago • 4 comments

请教一下，pytorch ocr crnn的resnet34 pth文件怎么转onnx？

nissansz avatar Feb 08 '23 03:02 nissansz

构建好网络加载好权重后，用torch.onnx.export就可以转了

frotms avatar Feb 08 '23 03:02 frotms

怎么构建？有demo 代码吗？

nissansz avatar Feb 08 '23 03:02 nissansz

按文档能推理起来，在推理代码后面加这一句就可以了

frotms avatar Feb 08 '23 03:02 frotms

infer可以成功，但是加上torch.onnx.export 报错

nissansz avatar Feb 08 '23 03:02 nissansz