onnx/onnx-tensorflow

What it solves

這項專案為使用以 ONNX (Open Neural Network Exchange) 格式建立的機器學習模型提供了一種方式，使其能夠在 TensorFlow 生態系統中執行。它解決了框架間的互操作性問題，讓使用者能夠將從其他匯出至 ONNX 的框架所產生的模型遷移到 TensorFlow 進行推理。

How it works

此工具會將 ONNX 模型轉換為 TensorFlow 模型表示法。轉換完成後，模型會被委派給 TensorFlow 執行以產生最終輸出。這可以透過命令列介面 (CLI) 或透過 Python 程式碼進行程式化操作。

Who it’s for

對於擁有 ONNX 格式模型並需要使用 TensorFlow 的執行引擎和硬體優化來執行它們的開發者和機器學習工程師。

Highlights

• 使用 onnx-tf convert 進行快速模型轉換的命令列介面。
• 支援透過 prepare 函數進行程式化轉換和直接推理。
• 與 ONNX Model Zoo 集成，以驗證各種模型的轉換成功率。
• 與執行相反操作 (TensorFlow 轉 ONNX) 的 tf2onnx 有明確的區別。