Add reshape layer after VariadicSplit function

PINTO0309 / openvino2tensorflow

This script converts the ONNX/OpenVINO IR model to Tensorflow's saved_model, tflite, h5, tfjs, tftrt(TensorRT), CoreML, EdgeTPU, ONNX and pb. PyTorch (NCHW) -> ONNX (NCHW) -> OpenVINO (NCHW) -> openvino2tensorflow -> Tensorflow/Keras (NHWC/NCHW) -> TFLite (NHWC/NCHW). And the conversion from .pb to saved_model and from saved_model to .pb and from .pb to .tflite and saved_model to .tflite and saved_model to onnx. Support for building environments with Docker. It is possible to directly access the host PC GUI and the camera to verify the operation. NVIDIA GPU (dGPU) support. Intel iHD GPU (iGPU) support.

MIT License

334 stars 40 forks source link

Add reshape layer after VariadicSplit function #80

Closed AlexisPapaioannou closed 2 years ago

AlexisPapaioannou commented 2 years ago

Can I add a reshape layer in the xml code after of a VariadicSplit function?

PINTO0309 commented 2 years ago

That's not possible with this tool. You can add it before ValiadicSplit and just replace Const in some layers behind it.

AlexisPapaioannou commented 2 years ago

Okay, thanks again!