Help Needed: AttributeError in tf2onnx Conversion from ONNX to TensorFlow Model

[orikkwak]

Help Needed: AttributeError in tf2onnx Conversion from ONNX to TensorFlow Model

[sawantkumar]

Hi @orikkwak , we're wondering if you may be able to resolve your issue by using AI-Edge-Torch, you can find more information here: googleblog. So instead of converting your model to onnx and then to tflite, you can directly convert your model from pytorch to tflite.

I have actually created a simple script for converting a resnet model here:

import torch
import torchvision
import ai_edge_torch

mobilenet_model = torchvision.models.resnet50()
sample_inputs = (torch.randn(1, 3, 224, 224),)

edge_model = ai_edge_torch.convert(mobilenet_model.eval(), sample_inputs)
edge_model.export("resnet.tflite")

If you want to, you can actually try visualizing the result in model-explorer as well.

Please try them out and let us know if this resolves your issue. If you still need further help, feel free to open a new issue at the respective repo.

[github-actions]

This issue is stale because it has been open for 7 days with no activity. It will be closed if no further activity occurs. Thank you.