Closed reflelia closed 1 year ago
I am making a tool that is easier to handle than openvino2tensorflow. Please read the README and the issue carefully and use this tool to convert it. It appears that your model contains several 5D Transpose, so you will need to adjust the behavior of the tool slightly.
New Tool - TFLite can be generated directly from ONNX https://github.com/PINTO0309/onnx2tf
Helpful Issue https://github.com/PINTO0309/onnx2tf/issues?q=is%3Aissue+label%3A%22Parameter+replacement%22
Thank you!
Issue Type
Bug
OS
Ubuntu
OS architecture
x86_64
Programming Language
Python
Framework
OpenVINO
Download URL for ONNX / OpenVINO IR
https://drive.google.com/file/d/1O4yY7BOuOaUKxpYpcC1GptySI81rq5pC/view?usp=share_link
Convert Script
openvino2tensorflow --model_path output-sim.xml --output_saved_model --output_pb --output_float16_quant_tflite --output_no_quant_float32_tflite --non_verbose
Description
I am trying to create a lightweight model for soft gated skip connection and convert it to tflite. reference -> https://github.com/dkurzend/Soft_Gated_Pose_Estimation_Pytorch In this process, shufflenetv2 that uses channel split was introduced, and the code for splitting channels is as follows.
split = x.split(x.shape[1]//2, dim=1) split1 = split[0] split2 = split[1]
Assuming that the first dimension of tensor is a channel, a separation operation was performed, but a KeyError occurred in the process of converting to tensorflow. I tried various methods of 6-7, such as transpose in the previous step of the VariadicSplit layer, but I couldn't solve it, so I leave a question.Relevant Log Output
Source code for simple inference testing code
No response