Sih NSSKJr SSKJr SSKrSSKJrJr SSjrg)) annotations)AnyN) ModelProtoValueInfoProtocV^ [5m SSjnU"T URR5 U"T URR5 U"T URR5 SU 4SjjnURRH0nUR nXn[ U5HupU"XYX5 M M2 URRH0n U R n X+n [ U 5HupU"XX5 M M2 [RRU5 U$)aThis function updates the dimension sizes of the model's inputs and outputs to the values provided in input_dims and output_dims. if the dim value provided is negative, a unique dim_param will be set for that dimension. Example. if we have the following shape for inputs and outputs: * shape(input_1) = ('b', 3, 'w', 'h') * shape(input_2) = ('b', 4) * shape(output) = ('b', 'd', 5) The parameters can be provided as: :: input_dims = { "input_1": ['b', 3, 'w', 'h'], "input_2": ['b', 4], } output_dims = { "output": ['b', -1, 5] } Putting it together: :: model = onnx.load('model.onnx') updated_model = update_inputs_outputs_dims(model, input_dims, output_dims) onnx.save(updated_model, 'model.onnx') cUHinURRRnURH6nUR S5(dMUR UR 5 M8 Mk g)N dim_param)type tensor_typeshapedimHasFieldaddr ) dim_param_set value_infosinfor r s [/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/tools/update_model_dims.pyinit_dim_param_set6update_inputs_outputs_dims..init_dim_param_set1sR DII))//Eyy<< ,,!%%cmm4! c >URRRRUn[ U[ 5(aUS:aOUR S5(a2URU:wa"[SUSUSUSURS3 5eXlgUS-[U5-nUT;a[S USUS 35eXTl g[ U[5(aXl g[S [U535e) Nr dim_valuez!Unable to set dimension value to z for axis z of z'. Contradicts existing dimension value ._z-Unable to generate unique dim_param for axis z,. Please manually provide a dim_param value.z@Only int or str is accepted as dimension value, incorrect type: ) r r r r isinstanceintrr ValueErrorstrr TypeError)tensorr jname dim_protogenerated_dim_paramrs r update_dim.update_inputs_outputs_dims..update_dim>s'KK++1155a8 c3  ax%%k22y7J7Jc7Q$;C5 1#TRVQWW~@I@S@STTUV'*#&*Sj3q6&9#&-7$Gs$tfUAB':# S ! !"% RSWX[S\R]^ r)rzset[str]rzlist[ValueInfoProto]returnNone) r rr rr!rr"rr'r() setgraphinputoutput value_infor" enumerateonnxchecker check_model)model input_dims output_dimsrr%input_ input_name input_dim_arrr!r r, output_nameoutput_dim_arrrs @rupdate_inputs_outputs_dimsr: s F"eM55.B5 5}ekk&7&78}ekk&8&89}ekk&<&<=.++##[[ ".  .FA vA 2/$ ++$$kk $1/FA vA 30%  LLU# Lr)r2rr3dict[str, list[Any]]r4r;r'r) __future__rtypingr onnx.checkerr/rrr:rrr@sA#+V V$V&V Vr