Media-Smart / vedastr

A scene text recognition toolbox based on PyTorch
Apache License 2.0
535 stars 100 forks source link

Export model into onnx #68

Open bharatsubedi opened 3 years ago

bharatsubedi commented 3 years ago

Thank you for your great work. how to convert 'TPS-ResNet-BiLSTM-Attention' module into onnx format?

bharatsubedi commented 3 years ago

who wants to convert cstr.pth model into onnx contact me I will help you.

chibohe commented 3 years ago

Thank you for your great work. how to convert 'TPS-ResNet-BiLSTM-Attention' module into onnx format?

Hi, have you successfully converted attention based models, thx.

mrljwlm commented 3 years ago

Thank you for your great work. how to convert 'TPS-ResNet-BiLSTM-Attention' module into onnx format?

+1 have you successfully converted this model? i meet this error: Exporting the operator grid_sampler to ONNX opset version 12 is not supported. Please open a bug to request ONNX export support for the missing operator.