paddlepaddle模型转onnx

1. 先将paddlepaddle训练模型转为推理模型
paddlepaddle的训练模型包含.pdparams文件和.pdopt文件,.pdparams文件为模型权重,.pdopt文件存放用于断点训练的参数;
利用paddledetection的tools目录中的export_model.py,将paddlepaddle的训练模型转为推理模型。
转换后的推理模型如下图所示:
paddlepaddle模型转onnx_第1张图片
2. 安装paddle2onnx

pip install paddle2onnx

3. 利用paddle2onnx转换模型

# 在cmd命令中执行如下操作
paddle2onnx  --model_dir E:\wyk\PaddleDetection-release-2.3\output_inference\lite_hrnet_30_384x288_coco --model_filename model.pdmodel --params_filename model.pdiparams --opset_version 11 --save_file test.onnx

paddlepaddle模型转onnx_第2张图片

转换onnx例子,如下图所示。
paddlepaddle模型转onnx_第3张图片

你可能感兴趣的:(paddlepaddle,深度学习,onnx)