B
    kndh                 @   s   d dl mZ d dlZd dlZd dlmZ dd Zedkre Zd dl	m
Z i ZejdkrbeejZeejeje edej dS )	    )absolute_importN)loggingc              C   s@   t  } | jdddd | jdddd | jddd	d
 |  S )Nz--input_modelTz"The path of input onnx model file.)requiredhelpz--output_modelz1The file path to write optimized onnx model file.z--input_shape_dict zgThe shape infos of inputs, e.g --input_shape_dict="{'image': [1, 3, 608, 608], 'scale_factor': [1, 2]}")defaultr   )argparseArgumentParseradd_argument
parse_args)parser r   8/tmp/pip-unpacked-wheel-qdbcl8o8/paddle2onnx/optimize.pyparse_arguments   s    r   __main__r   zModel optmized, saved in {}.)
__future__r   r   sysZpaddle2onnx.utilsr   r   __name__argsZ%paddle2onnx.paddle2onnx_cpp2py_exportZpaddle2onnx_cpp2py_exportZc_p2oZ
shape_dictZinput_shape_dictevaloptimizeZinput_modelZoutput_modelinfoformatr   r   r   r   <module>   s   

