Paddle2ONNX
Paddle2ONNX copied to clipboard
Paddle导出onnx模型
这是我使用导出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
我想导出成下面格式的onnx,输出和yolo的输出是一样的。