FangShancheng / ABINet

Read Like Humans: Autonomous, Bidirectional and Iterative Language Modeling for Scene Text Recognition
Other
421 stars 72 forks source link

Support of the script to export to ONNX model #59

Open rose-jinyang opened 2 years ago

rose-jinyang commented 2 years ago

Hello How are you? Thanks for contributing to this project. I have some troubles in exporting your pre-trained torch model to ONNX model. Could u provide the script for exporting to ONNX model? Thanks

bbsvip commented 1 year ago

I edited it a bit to be able to convert to onnx. You can refer to here