tensorflow-onnx
tensorflow-onnx copied to clipboard
Feat : support NHWC to NCHW in split node wo transpose
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_outputsfunctions 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.
-
in unit_test,
python setup.py test, i got results as below. i think those failed issues are not related in my code.