Closed PetiteFleurPF closed 3 months ago
current schema of the model FLOAT32 which generates the error.
(node 276 - FLOAT32 - Add - on the down of the second picture: It makes sense for the error to appear because we can see that we're trying to add two matrices of different dimensions. (1x4) and (1x1))
(A large scale of the equivalent of ONNX version)
Log of ONNX conversion
Log of validation
(the first line is the gather component which can be helpfull to localize the bug)
I will not investigate unless you share the ONNX files with me.
It's usually because of the garbage-like post-processing.
@PINTO0309 Hi PINTO0309 - I updated my description with a link to download my model :)
Post-processing is too redundant. There are no shapes that can be estimated by the tool after NonZero
. Models in which the number of output elements varies with the content of the input data cannot be accurately transformed by onnx2tf. This is not a bug.
Cut out the model above the red line.
Theoretically I can't modify my model, so if I understand correctly the solution is not to modify your code but my model? Thank you in advance for your time.
if I understand correctly the solution is not to modify your code but my model?
Yes. Aside from the fact that it can and cannot be converted, the model is too redundant and inefficient. In particular, all models generated without modification of Detectron / Detectron2
and others will generate this kind of redundant post-processing.
The post-processing of your model can be replaced by the process shown below.
OK many thanks :)
Question - what tool do you recommend to do it, please? ONNX surgeon?
https://github.com/PINTO0309/PINTO_model_zoo/tree/main/307_YOLOv7 https://github.com/PINTO0309/PINTO_model_zoo/tree/main/322_YOLOv7_Head https://github.com/PINTO0309/PINTO_model_zoo/tree/main/326_YOLOPv2 https://github.com/PINTO0309/PINTO_model_zoo/tree/main/334_DAMO-YOLO https://github.com/PINTO0309/PINTO_model_zoo/tree/main/337_FreeYOLO https://github.com/PINTO0309/PINTO_model_zoo/tree/main/356_EdgeYOLO https://github.com/PINTO0309/PINTO_model_zoo/tree/main/447_YOLOX-Wholebody-with-Wheelchair
What I meant was a tool that lets me modify my model to change, replace or delete parts of the post-processing ^^ Sorry for the misunderstanding.
Issue Type
Others
OS
Linux
onnx2tf version number
1.21.0
onnx version number
1.15.0
onnxruntime version number
1.17.1
onnxsim (onnx_simplifier) version number
0.4.33
tensorflow version number
2.16.1
Download URL for ONNX
model.onnx.zip
Parameter Replacement JSON
Description
INFO: 283 / 301 INFO: onnx_op_type: Gather onnx_op_name: /Gather_8 INFO: input_name.1: /GatherND_1_output_0 shape: ['unk211', 4] dtype: float32 INFO: input_name.2: /TopK_output_1 shape: ['unk327'] dtype: int64 INFO: output_name.1: /Gather_8_output_0 shape: ['unk327', 4] dtype: float32 INFO: tf_op_type: gather_v2 INFO: input.1.params: name: tf.compat.v1.gather_nd_1/GatherNd:0 shape: (None, 4) dtype: <dtype: 'float32'> INFO: input.2.indices: name: tf.operators__.add_30/AddV2:0 shape: (None,) dtype: <dtype: 'int64'> INFO: input.3.axis: val: 0 INFO: output.1.output: name: tf.compat.v1.gather_3/GatherV2:0 shape: (None, 4) dtype: <dtype: 'float32'>