hl nSSKJrJrJrJr SSKrSSKJrJr S\S\\ \\4S\\ \\4S\4Sjr g) )AnyDictListSetN) ModelProtoValueInfoProtomodel input_dims output_dimsreturnc ^ [5m S[[S[[SS4SjnU"T UR R 5 U"T UR R5 U"T UR R5 S[S[S[S [SS4 U 4S jjnUR R H0nURnXn[U5HupU"XYX5 M M2 UR RH0n U Rn X+n [U 5HupU"XX5 M M2 [RRU5 U$) a This 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') dim_param_set value_infosr NcUHinURRRnURH6nUR S5(dMUR UR 5 M8 Mk g)N dim_param)type tensor_typeshapedimHasFieldaddr)rrinforrs V/var/www/fran/franai/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! tensorrjnamec >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: ) rrrr isinstanceintrr! ValueErrorstrr)rrrr 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)setrr'rrgraphinputoutput value_inforr%r enumerateonnxchecker check_model)r r r rr*input_ input_name input_dim_arrrrr/ output_nameoutput_dim_arrrs @rupdate_inputs_outputs_dimsr: s6H"eM53x5.2>.B5 5}ekk&7&78}ekk&8&89}ekk&<&<=>34.++##[[ ".  .FA vA 2/$ ++$$kk $1/FA vA 30%  LLU# Lr) typingrrrr onnx.checkerr2rrr'r:rrr>s[ ('+W WS$s)^$Wc49n%W Wr