当前位置：网站首页>Paddlepaddle model to onnx

Paddlepaddle model to onnx

2022-04-23 03:35:00 【Crispy eggplant strips】

1. First the paddlepaddle Turn the training model into reasoning model
paddlepaddle The training model includes .pdparams Document and .pdopt file ,.pdparams The file is the model weight ,.pdopt The file stores the parameters used for breakpoint training ;
utilize paddledetection Of tools In the directory export_model.py, take paddlepaddle The training model is transformed into reasoning model .
The converted reasoning model is shown in the figure below ：
Insert picture description here
2. install paddle2onnx

pip install paddle2onnx

3. utilize paddle2onnx Transformation model

#  stay cmd Execute the following operations in the command 
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