search

VamosC / CLIP4STR

An implementation of "CLIP4STR: A Simple Baseline for Scene Text Recognition with Pre-trained Vision-Language Model".
Apache License 2.0
115 stars 14 forks source link

Convert to ONNX #9

Closed YenYunn closed 6 months ago

YenYunn commented 6 months ago

How can I convert a model to ONNX for inference implementation?

VamosC commented 6 months ago

Hi, @YenYunn

I am not familiar with this topic. Maybe you can refer to https://pytorch.org/tutorials/beginner/onnx/export_simple_model_to_onnx_tutorial.html https://huggingface.co/docs/transformers/serialization for a few suggestions.

Finally, if you do success, please seriously consider make a PR to this repo, thx!

YenYunn commented 6 months ago

Thanks for your reply. I will try it