Open gordinmitya opened 4 years ago
have you tried with tf=1.15 ? I believe thats the last fully supported according to the docs?
Because users often have their own preferences for which variant of Tensorflow to install (i.e., a GPU version instead of a CPU version), we do not explicitly require tensorflow in the installation script. It is therefore users' responsibility to ensure that the proper variant of Tensorflow is available to ONNX-TF. Moreoever, we require Tensorflow version == 1.15.0.
From Readme.md.
Also check out :
https://github.com/onnx/onnx-tensorflow/blob/master/Versioning.md
No idea if that holds up though.
Also, im about to try this exact conversion today, so ill let you know how it goes with TF 1.15
@vade Hey, thank you for the response!
I guess the doc is inconsistent a little bit. In installation section: "Install Tensorflow >= 2.0 and tensorflow-addons. (Note for Tensorflow 1.x please refer the tf-1.x branch)". Also I sow in commits "Support Tensorflow 2.0".
Don't remember exact error, but I've tried with 1.15 and it doesn't worked. So please text if you'll find the solution.
Same problem, no solution yet #617 There should be at least one working solution, however I couldn't find any so far
overall onnx-tf doesn't seem working anymore I was able to convert mobilenet_v2 with https://github.com/nerox8664/pytorch2keras. Convert to saved_model then to tflite.
Describe the bug
Unable to convert model from torchvision to pb/tflite. Conversion completes but output file can't be open in neutron app neither executed with tensowflow. If I continue conversion to tflite, it finishes without errors but still file is broken.
To Reproduce
Regular MobilenetV2 model obtained from torchvision as described in official doc link to onnx is below.
given python 3.7 pip install tensorflow-cpu tensorflow-addons onnx install onnx-tensorflow from git as described in readme
you can find Dockerfile and image here
ONNX model file
https://drive.google.com/open?id=1OtQfwWiS8BZ2qW3dOoSbHaaQrZbLMeHL
Python, ONNX, ONNX-TF, Tensorflow version
This section can be obtained by running
get_version.py
from util folder.Additional context