tensorflow-onnx icon indicating copy to clipboard operation
tensorflow-onnx copied to clipboard

Feat : support NHWC to NCHW in split node wo transpose

Open dongryeolle opened this issue 2 months ago • 0 comments

Summary

  • current tf2onnx doesn't support to convert without transpose NHWC to NCHW in split node.
  • and the most of functions are not consider when operator has multiple outputs.

Changes

  • add features that change axis of Split in onnx_opset for NHWC to NCHW.
  • add _switch_transpose_and_node_with_multiple_outputs functions for switching transpose node and split node has multiple-output.

Test

  • Unfortunately due to security issues, I attach before & after figure instead of my test model. image

  • in unit_test,python setup.py test, i got results as below. i think those failed issues are not related in my code.

image

dongryeolle avatar Oct 01 '25 02:10 dongryeolle