o i@sdddlZddlZddZdejdefddZedkr0eZeej Z ee e e ej dSdS) NcCs4t}|jdddd|jdddd|}|S)Nz--inputTz input model)requiredhelpz--outputz output model)argparseArgumentParser add_argument parse_args)parserargsr i/home/kim/smarthome/.venv/lib/python3.10/site-packages/onnxruntime/tools/remove_initializer_from_input.pyget_argss r modelreturncCsj|jdkr tddS|jj}i}|D]}|||j<qd}|jjD]}|j|vr2d}|||jq!|S)NzLModel with ir_version below 4 requires to include initializer in graph inputFT)Z ir_versionprintgraphinputname initializerremove)r ZinputsZ name_to_inputrmodifiedrr r r remove_initializer_from_inputs    r__main__) rZonnxr Z ModelProtoboolr__name__r loadrr saveoutputr r r r s