[YOLOX-X] [Faster-RCNN] [YOLOv7] TensorFlow aborts when exporting a model with NMS to Keras (.h5)

[PINTO0309]

Issue Type

Others

onnx2tf version number

1.5.29

onnx version number

1.13.0

tensorflow version number

2.10.0

Download URL for ONNX

https://s3.ap-northeast-2.wasabisys.com/temp-models/onnx2tf_146/yolox_x.onnx
https://github.com/PINTO0309/onnx2tf/releases/download/1.1.28/faster_rcnn-10.onnx
https://github.com/PINTO0309/onnx2tf/releases/download/1.1.28/yolov7_tiny_head_0.768_post_480x640.onnx

Parameter Replacement JSON

N/A

Description

1. Personal
2. TensorFlow aborts when exporting a model with NMS to Keras (.h5)

   onnx2tf -i yolox_x.onnx -oh5
   

   ValueError: This Keras op layer was generated from <function non_max_suppression at 0x7fd6a96ff1f0>,
   a method that is not publicly exposed in the TensorFlow API.
   This may have happened if the method was explicitly decorated to add dispatching support,
   and it was used during Functional model construction.
   To ensure cross-version compatibility of Keras models that use op layers,
   only op layers produced from public TensorFlow API symbols can be serialized.