无法将 pytorch 模型转换为 ONNX
Couldn't convert pytorch model to ONNX
我使用了这个仓库：https://github.com/Turoad/lanedet
将使用 mobilenetv2 作为 backbone 的 pytorch 模型转换为 ONNX，但我没有成功。
我收到一个运行时错误：
RuntimeError: Exporting the operator eye to ONNX opset version 12 is
not supported. Please open a bug to request ONNX export support for
the missing operator.
真的很失望，看着这个模型给出的好结果和它提供的快速性能，
有什么方法可以修复这个错误吗？因为我需要将其转换为 ONNX，然后转换为 TF lite 模型以在 Android App 中使用它
我将提供我使用过的预训练模型以及我在转换时遵循的方式..
非常感谢您的帮助！
我的 colab 笔记本：
https://colab.research.google.com/drive/18udIh8tNJvti7jKmR4jRaRO-oYDgRmvA?usp=sharing
我使用的预训练模型：
https://drive.google.com/file/d/1o3-BgLIQesurIyDCKGliqbo2inUA5cPw/view?usp=sharing
使用torch>=1.7.0转换模型，因为添加了操作Eye
我使用了这个仓库：https://github.com/Turoad/lanedet 将使用 mobilenetv2 作为 backbone 的 pytorch 模型转换为 ONNX，但我没有成功。
我收到一个运行时错误：
RuntimeError: Exporting the operator eye to ONNX opset version 12 is not supported. Please open a bug to request ONNX export support for the missing operator.
真的很失望，看着这个模型给出的好结果和它提供的快速性能，
有什么方法可以修复这个错误吗？因为我需要将其转换为 ONNX，然后转换为 TF lite 模型以在 Android App 中使用它 我将提供我使用过的预训练模型以及我在转换时遵循的方式..
非常感谢您的帮助！
我的 colab 笔记本：
https://colab.research.google.com/drive/18udIh8tNJvti7jKmR4jRaRO-oYDgRmvA?usp=sharing
我使用的预训练模型：
https://drive.google.com/file/d/1o3-BgLIQesurIyDCKGliqbo2inUA5cPw/view?usp=sharing
使用torch>=1.7.0转换模型，因为添加了操作Eye