search

zysNLP / transformers2TensorRT

本项目介绍了使用Transformers模块微调文本分类模型，以及模型转pth格式，进而转为onnx格式和TensorRT的engine格式，最后通过TensorRT进行推理的完整流程
7 stars 1 forks source link

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

Open nissansz opened 1 year ago

nissansz commented 1 year ago

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

zysNLP commented 1 year ago

一样的啊，请参考model2onxx.py

nissansz commented 1 year ago

是crnn，有字符集要加载，再导出onnx，不知道怎么改