Paddle2ONNX icon indicating copy to clipboard operation
Paddle2ONNX copied to clipboard

Paddle导出onnx模型

Open yuexiayiren159aaa opened this issue 2 years ago • 0 comments

这是我使用导出onnx的命令

python tools/export_model.py -c configs/ppyolo/ppyolo_tiny_650e_coco.yml
-o weights=weights/ppyolo_tiny_650e_coco.pdparams
TestReader.inputs_def.image_shape=[3,416,416]
--output_dir inference_model

paddle2onnx --model_dir inference_model/ppyolo_tiny_650e_coco/
--model_filename ppyolo_tiny_650e_coco/model.pdmodel
--params_filename ppyolo_tiny_650e_coco/model.pdiparams
--save_file inference_model/ppyolo_tiny_650e_coco/ppyolo_123.onnx
--opset_version 11
--enable_onnx_checker True
--input_shape_dict "{'image': [1, 3, 416, 416]}"

下面是我导出的onnx

image image

我想导出成下面格式的onnx,输出和yolo的输出是一样的。

image image

yuexiayiren159aaa avatar Apr 01 '22 03:04 yuexiayiren159aaa