paddle2onnx PaddlePaddle 到 ONNX 模型转换器开源项目

我要开发同款
匿名用户2020年02月18日
55阅读
开发技术Python
所属分类人工智能、机器学习/深度学习
授权协议Apache-2.0

作品详情

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

##CPU版本安装命令pipinstall-fhttps://paddlepaddle.org.cn/pip/oschina/cpupaddlepaddle##GPU版本安装命令pipinstall-fhttps://paddlepaddle.org.cn/pip/oschina/gpupaddlepaddle-gpu

参数说明

参数参数说明fluid_modelpaddlefluid模型和模型参数所在目录onnx_model转化成onnx模型的模型名称name_prefix[可选]某些paddle模型的模型参数加了前缀,则需要指定模型参数前缀,例如@HUB_mobilenet_v2_imagenet@conv6_2_expand_bn_scalefluid_model_name[可选]如果导入的paddle模型不是默认__model__,需要指定模型的名字fluid_params_name[可选]如果导入的paddle模型参数是合并在一个文件里面,需要指定模型参数文件名debug[可选]如果开发者要对转化的模型进行精度测试,打开此开关return_variable[可选]在debug模式中,如果paddle模型返回的结果是LoDTensor,需要打开此开关check_task[可选]在debug模式中,根据不同配置项选择不同的执行器和数据构造器image_path[可选]在debug模式中,可以选择加载不同的图片进行精度验证
声明:本文仅代表作者观点,不代表本站立场。如果侵犯到您的合法权益,请联系我们删除侵权资源!如果遇到资源链接失效,请您通过评论或工单的方式通知管理员。未经允许,不得转载,本站所有资源文章禁止商业使用运营!
下载安装【程序员客栈】APP
实时对接需求、及时收发消息、丰富的开放项目需求、随时随地查看项目状态

评论