PaddlePaddle 到 ONNX 模型转换器 paddle2onnx

Apache-2.0
Python 查看源码»
跨平台
百度
2020-02-18
编辑部的故事

疫情之下,程序员如何转型?共享数字化转型加速度,就在微软在线技术峰会>>>

paddle2onnx 是一个 PaddlePaddle 到 ONNX 模型转换器,支持将 PaddlePaddle 框架下产出的模型转化到 ONNX 模型格式。

参数说明

参数 参数说明
fluid_model paddle fluid模型和模型参数所在目录
onnx_model 转化成onnx模型的模型名称
name_prefix [可选]某些paddle模型的模型参数加了前缀,则需要指定模型参数前缀,例如@HUB_mobilenet_v2_imagenet@conv6_2_expand_bn_scale
fluid_model_name [可选]如果导入的paddle模型不是默认__model__,需要指定模型的名字
fluid_params_name [可选]如果导入的paddle模型参数是合并在一个文件里面,需要指定模型参数文件名
debug [可选]如果开发者要对转化的模型进行精度测试,打开此开关
return_variable [可选]在debug模式中,如果paddle模型返回的结果是LoDTensor,需要打开此开关
check_task [可选]在debug模式中,根据不同配置项选择不同的执行器和数据构造器
image_path [可选]在debug模式中,可以选择加载不同的图片进行精度验证
的码云指数为
超过 的项目
加载中

评论(0)

暂无评论

暂无资讯

暂无问答

暂无博客

返回顶部
顶部