None in graph_node_input.shape

[MichaelMonashev]

Issue Type

Others

OS

Linux

onnx2tf version number

1.25.12

onnx version number

1.16.2

onnxruntime version number

1.19.2

onnxsim (onnx_simplifier) version number

0.4.36

tensorflow version number

2.17.0

Download URL for ONNX

https://drive.google.com/file/d/1PlXNBPGgNLyy-MKJJRe-yzakSZtJCtT6/view?usp=sharing

Parameter Replacement JSON

{
  "format_version": 1,
  "operations": []
}

Description

$ onnx2tf -i yolov7x.onnx

...

ERROR: The trace log is below.
Traceback (most recent call last):
  File "/home/xxx/.local/lib/python3.12/site-packages/onnx2tf/utils/common_functions.py", line 312, in print_wrapper_func
    result = func(*args, **kwargs)
             ^^^^^^^^^^^^^^^^^^^^^
  File "/home/xxx/.local/lib/python3.12/site-packages/onnx2tf/utils/common_functions.py", line 385, in inverted_operation_enable_disable_wrapper_func
    result = func(*args, **kwargs)
             ^^^^^^^^^^^^^^^^^^^^^
  File "/home/xxx/.local/lib/python3.12/site-packages/onnx2tf/utils/common_functions.py", line 55, in get_replacement_parameter_wrapper_func
    func(*args, **kwargs)
  File "/home/xxx/.local/lib/python3.12/site-packages/onnx2tf/ops/ArgMax.py", line 69, in make_node
    tensor_rank=len(graph_node_input.shape),
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^
TypeError: object of type 'NoneType' has no len()

ERROR: input_onnx_file_path: yolov7x.onnx
ERROR: onnx_op_name: wa/end2end/ArgMax
ERROR: Read this and deal with it. https://github.com/PINTO0309/onnx2tf#parameter-replacement
ERROR: Alternatively, if the input OP has a dynamic dimension, use the -b or -ois option to rewrite it to a static shape and try again.
ERROR: If the input OP of ONNX before conversion is NHWC or an irregular channel arrangement other than NCHW, use the -kt or -kat option.
ERROR: Also, for models that include NonMaxSuppression in the post-processing, try the -onwdt option.

[MichaelMonashev]

Command to genarate yolov7x.onnx: python3 export.py --weights yolov7x.pt --grid --end2end --simplify --topk-all 100 --iou-thres 0.65 --conf-thres 0.35 --img-size 640 640 --max-wh 640 --dynamic-batch

Error occur with --dynamic-batch argument only. With fixed batch size onnx2tf works without errors.

export.py from https://github.com/WongKinYiu/yolov7/ yolov7x.pt from https://github.com/WongKinYiu/yolov7/releases/download/v0.1/yolov7x.pt