B knd' @sddlmZddlmZddlZddlZddlZddlZddl m Z ddZ ddZ dddZ dddZddZedkr~edS))absolute_import) text_typeN)loggingcCs(t|dkrdS|dd}t|}|S)Nr )lenreplaceeval)vr 7/tmp/pip-unpacked-wheel-qdbcl8o8/paddle2onnx/command.pystr2lists   r c CsRt}|jddtddd|jddtddd|jdd tdd d|jd d tdd d|jddtddd|jddtddd|jdtjddd|jddtdddddgdd |jd!td"d#d|jd$tjdd%d|jd&tjd'd(d|jd)d*d+d'd,d-|jd.d/tdd0d|jd1tjdd2d|jd3tdd4d|jd5tjd'd6d|S)7Nz --model_dirz-mzxPaddlePaddle model directory, if params stored in single file, you need define '--model_filename' and 'params_filename'.)typedefaulthelpz--model_filenamez-mfzRPaddlePaddle model's network file name, which under directory seted by --model_dirz--params_filenamez-pfzvPaddlePaddle model's param file name(param files combined in single file), which under directory seted by --model_dir.z --save_filez-szfile path to save onnx modelz--opset_versionz-ov z set onnx opset version to exportz--input_shape_dictz-isdNonezdefine input shapes, e.g --input_shape_dict="{'image':[1, 3, 608, 608]}" or--input_shape_dict="{'image':[1, 3, 608, 608], 'im_shape': [1, 2], 'scale_factor': [1, 2]}"z--enable_dev_versionTzQwhether to use new version of Paddle2ONNX which is under developing, default Truez--deploy_backendz-d onnxruntimeZtensorrtZrknnZothersz3Quantize model deploy backend, default onnxruntime.)rrchoicesrz--save_calibration_filezcalibration.cachezEThe calibration cache for TensorRT deploy, default calibration.cache.z--enable_onnx_checkerz/whether check onnx model validity, default Truez--enable_paddle_fallbackFz:whether use PaddleFallback for custom op, default is Falsez --versionz-v store_truezget version of paddle2onnx)actionrrz--output_namesz-onzdefine output names, e.g --output_names="["output1"]" or --output_names="["output1", "output2", "output3"]" or --output_names="{"Paddleoutput":"Onnxoutput"}"z--enable_auto_update_opsetz1whether enable auto_update_opset, default is Truez--external_filenamez?The filename of external_data when the model is bigger than 2G.z--export_fp16_modelz4Whether export FP16 model for ORT-GPU, default False)argparseArgumentParser add_argument _text_typeintast literal_evalr )parserr r r arg_parser s rrTrFc Cs`ddlm} | ||||||||i| | | | }|dk rXt|d}||WdQRXn|SdS)Nrwb)Z%paddle2onnx.paddle2onnx_cpp2py_exportZpaddle2onnx_cpp2py_exportexportopenwrite) model_file params_file save_file opset_versionauto_upgrade_opsetverboseenable_onnx_checkerenable_experimental_openable_optimizedeploy_backendcalibration_file external_fileexport_fp16_modelZc_p2oZonnx_model_strfr r r c_paddle_to_onnxs    r3rONNXc Cs0tdddlm} | |||||||||| S)Nz[Deprecated] `paddle2onnx.command.program2onnx` will be deprecated in the future version, the recommended usage is `paddle2onnx.export`r) program2onnx)rwarningZpaddle2onnx.legacy.commandr5) model_dirr'model_filenameparams_filenamer(r+operator_export_typeinput_shape_dict output_namesauto_update_opsetr5r r r r5s   r5c Csttjdkr0tdtdtddSt}|}|jrbddl}td |j dS|j dk stt d|j dk st dt|j}d }|jrtd d }|jdk r|jrtd t|jttfstd t|j|dk r|jrtd|jrtj|j |j}|jdkr(d}ntj|j |j}|jdkrLd|_tj|j }|rxtj |sxt!|tj||j}|j"} t#|||j |j$|j%d|j&dd|j'| ||j(d tdtddSt)|j |j |j|j|j$|j&|||j|j%d tdtddS)Nz2Use "paddle2onnx -h" to print the help informationz:For more information, please follow our github repo below:z7Github: https://github.com/PaddlePaddle/paddle2onnx.gitrz4paddle2onnx-{} with python>=3.6, paddlepaddle>=2.0.0zD--model_dir should be defined while translating paddle model to onnxzD--save_file should be defined while translating paddle model to onnxr4z{[Deprecated] The flag `--enable_paddle_fallback` will be deprecated, and only works while `--enable_dev_version False` now.ZPaddleFallbackz[Deprecated] The flag `--output_names` is deprecated, if you need to modify the output name, please refer to this tool https://github.com/jiangjiajun/PaddleUtils/tree/main/onnx zEThe output_names should be 'list' or 'dict', but received type is %s.z[Deprecated] The flag `--input_shape_dict` is deprecated, if you need to modify the input shape of PaddlePaddle model, please refer to this tool https://github.com/jiangjiajun/PaddleUtils/tree/main/paddle rZ external_dataT) r%r&r'r(r)r*r+r,r-r.r/r0r1z8===============Make PaddlePaddle Better!================z8A little survey: https://iwenjuan.baidu.com/?code=r8hu2s)r(r+r:r;r<r=)*rsysargvrinfor parse_argsversion paddle2onnxformat __version__r7AssertionErrorr'r r;Zenable_paddle_fallbackr6r<Zenable_dev_version isinstancelistdict TypeErrorrospathjoinr8r9Zexternal_filenamedirnameexistsmkdirZsave_calibration_filer3r(Zenable_auto_update_opsetr+r.r1r5) rargsrDr;r:r%r& base_pathr0r/r r r mains            rT__main__) rNr TTTTTrrrF)NNrFr4NNT) __future__rsixrrrrr?rLZpaddle2onnx.utilsrr rr3r5rT__name__r r r r s>   h  \