WebAI.js icon indicating copy to clipboard operation
WebAI.js copied to clipboard

模型转换问题

Open alwan1989 opened this issue 3 years ago • 3 comments

作者你好啊,感谢您开源这么优秀的项目,但我有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?可以说明下相应的处理流程嘛?万分感谢……

alwan1989 avatar Sep 21 '22 01:09 alwan1989

作者你好啊,感谢您开源这么优秀的项目,但我有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,不过之后需要手动配置一下模型预处理的一些配置

jm12138 avatar Sep 21 '22 01:09 jm12138

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

alwan1989 avatar Sep 21 '22 01:09 alwan1989

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

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

jm12138 avatar Sep 21 '22 02:09 jm12138