Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

模型转换问题 #5

Open
alwan1989 opened this issue Sep 21, 2022 · 3 comments
Open

模型转换问题 #5

alwan1989 opened this issue Sep 21, 2022 · 3 comments

Comments

@alwan1989
Copy link

作者你好啊,感谢您开源这么优秀的项目,但我有2个疑问,想请教下……疑问1:在项目中你所用的都是paddle的预训练模型,是用paddlejsconverter --modelPath=user_model_path --paramPath=user_model_params_path --outputDir=model_saved_path --useGPUOpt=True 这个转换成你的model.onnx.json的嘛?如果不是,可以提供一个简易的转换脚本供参考嘛?……疑问2:是不是可以通过torch转成onnx,然后也可以在onnxruntime里面找转换脚本,转成你所需的model.onnx.json?可以说明下相应的处理流程嘛?万分感谢……

@jm12138
Copy link
Member

jm12138 commented Sep 21, 2022

作者你好啊,感谢您开源这么优秀的项目,但我有2个疑问,想请教下……疑问1:在项目中你所用的都是paddle的预训练模型,是用paddlejsconverter --modelPath=user_model_path --paramPath=user_model_params_path --outputDir=model_saved_path --useGPUOpt=True 这个转换成你的model.onnx.json的嘛?如果不是,可以提供一个简易的转换脚本供参考嘛?……疑问2:是不是可以通过torch转成onnx,然后也可以在onnxruntime里面找转换脚本,转成你所需的model.onnx.json?可以说明下相应的处理流程嘛?万分感谢……

其实就是普通的onnx模型文件,后缀是json只是因为方便传输而已,torch也可以导出onnx,不过之后需要手动配置一下模型预处理的一些配置

@alwan1989
Copy link
Author

好的,感谢……就是导出onnx,再simple后,把模型model.onnx手动改成后缀model.onnx.json就行了是吧?你用的paddle预训练模型也是转成onnx的嘛?

@jm12138
Copy link
Member

jm12138 commented Sep 21, 2022

好的,感谢……就是导出onnx,再simple后,把模型model.onnx手动改成后缀model.onnx.json就行了是吧?你用的paddle预训练模型也是转成onnx的嘛?

项目里面的预训练模型都是paddle 导出的 onnx 模型文件,其实 onnxruntime 可以直接加载的。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants