paddle2onnxPaddlePaddle 到 ONNX 模型转换器
paddle2onnx 是一个 PaddlePaddle 到 ONNX 模型转换器，支持将 PaddlePaddle 框架下产出的模型转化到 ONNX 模型格式。
## CPU版本安装命令
pip install -f https://paddlepaddle.org.cn/pip/oschina/cpu paddlepaddle
## GPU版本安装命令
pip install -f https://paddlepaddle.org.cn/pip/oschina/gpu paddlepaddle-gpu
参数说明
| 参数 | 参数说明 |
|---|---|
| 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模式中，可以选择加载不同的图片进行精度验证 |
评论