如何将pytorch训练好的dbnet模型转化为paddle模型

[yulm-1]

感谢您参与 X2Paddle 社区！ 问题模版为了 X2Paddle 能更好的迭代，例如新功能发布、 RoadMaps 和错误跟踪. :smile_cat:

问题描述

• 错误信息
• 错误截图

具体信息

想将pytorch训练好的dbnet模型转化为paddle模型进行后续的和文本识别串联的工作

• 转换模型后用处

  • [ ] 使用 Paddle 框架/ PaddleInference 推理预测

• 模型来源 https://github.com/MhLiao/DB自己训练的模型

• 应用场景

• 版本信息 PaddlePaddle => :grey_question: X2Paddle => :grey_question: 来源框架版本(PyTorch) => :grey_question:

• 您的联系方式(邮箱/微信/电话)15133731918

[wjj19950828]

@yulm-1 torch模型有两种转换方式

1、直接Torch2Paddle，参考这个文档：https://github.com/PaddlePaddle/X2Paddle/blob/develop/docs/inference_model_convertor/pytorch2paddle.md

2、转成ONNX，再ONNX2Paddle，然后运行下面命令：

x2paddle --framework=onnx --model=model.onnx --save_dir=pd_model

建议都试下