1. 先将paddlepaddle训练模型转为推理模型
paddlepaddle的训练模型包含.pdparams文件和.pdopt文件,.pdparams文件为模型权重,.pdopt文件存放用于断点训练的参数;
利用paddledetection的tools目录中的export_model.py,将paddlepaddle的训练模型转为推理模型。
转换后的推理模型如下图所示:
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
转换onnx例子,如下图所示。