Paddle导出onnx时候去掉后处理

[yuexiayiren159aaa]

使用 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 inference_model/ppyolo_tiny_650e_coco/model.pdmodel
--params_filename inference_model/ppyolo_tiny_650e_coco/model.pdiparams
--save_file inference_model/ppyolo_tiny_650e_coco/ppyolo3.onnx
--enable_onnx_checker True

我想导出成下main的格式onnx

[jiangjiajun]

https://github.com/jiangjiajun/PaddleUtils/tree/main/onnx 参考这里的onnx模型裁剪脚本