ONNX：開放神經網路交換

阿新 • • 發佈：2019-01-16

參考：

開放式神經網路交換（ONNX）是邁向開放式生態系統的第一步，使AI開發人員能夠隨著專案發展而選擇正確的工具。 ONNX為AI模型提供了一個開源格式。它定義了一個可擴充套件的計算圖模型，以及內建運算子和標準資料型別的定義。最初專注於推理（評估）所需的功能。

Caffe2，PyTorch，Microsoft Cognitive Toolkit，Apache MXNet和其他工具正在開發ONNX支援。實現不同框架之間的互操作性並簡化從研究到生產的路徑將增加AI社群的創新速度。 ONNX處於早期階段，邀請社群提交反饋並幫助進一步發展ONNX。

總的來說，ONNX 是微軟聯合多個相關公司一起推動的深度學習模型標準

。該標準可以更好地為軟體和硬體廠商間提供更好的可互動性。多家公司的支援也利於推動不同框架內部的互操作支援。MMdnn也將支援ONNX格式

#安裝（支援conda，以在conda下安裝為例：）

1.用conda-forge建立分支：

conda install -c conda-forge onnx

2.從 PyPi 安裝 ONNX（注意：設定 environment variable ONNX_ML=1 for onnx-ml）

pip install onnx

python setup.py install

注意：當在非anaconda環境安裝時，在執行pip安裝ONNX之前，要確保已經安裝了 Protobuff編譯器，例如，以Ubuntu為例：

sudo apt-get install protobuf-compiler libprotoc-dev
pip install onnx

3.安裝完成後，執行：

python -c "import onnx"

來驗證能否工作。請注意，該命令在 source checkout directory 中不起作用; 在這種情況下，你會看到：

ModuleNotFoundError: No module named 'onnx.onnx_cpp2py_export'

此時需要更換另一路徑來修復錯誤

#測試

ONNX用作為測試驅動器。為了能執行測試，需要安裝pytest：

pip install pytest-cov nbval

pytest完成安裝後，執行：

pytest

轉載註明出處！

編輯