ONNX(英語:Open Neural Network Exchange)是一种针对机器学习所设计的开放式的文件格式,用于存储训练好的模型。它使得不同的人工智能框架(如Pytorch、MXNet)可以采用相同格式存储模型数据并交互。 ONNX的规范及代码主要由微软亚马逊FacebookIBM等公司共同开发,以开放源代码的方式托管在Github上。[2][3][4]  目前官方支持加载ONNX模型并进行推理的深度学习框架有: Caffe2, PyTorch, MXNet,ML.NET,TensorRT 和 Microsoft CNTK,并且 TensorFlow 也非官方的支持ONNX。

Open Neural Network Exchange (ONNX)
原作者Facebook, Microsoft
開發者Linux基金会
首次发布2017年9月,​7年前​(2017-09
当前版本1.17.0[1](2024年10月1日)
源代码库 編輯維基數據鏈接
编程语言C++, Python
操作系统Windows, Linux
类型人工智能生态圈
许可协议最初为MIT许可证
后改为Apache许可证2.0
网站onnx.ai/ 编辑维基数据

历史

ONNX最初被命名为Toffee[5],并且由脸书内发布PyTorch的团队开发。[6] 在2017年,ONNX第一个正式版本由脸书与微软共同发布。在此之后,IBM, 华为, 英特尔, AMD, Arm高通 都宣布计划支持ONNX。

参考文献

  1. ^ Release 1.17.0. 2024年10月1日 [2024年10月22日]. 
  2. ^ onnx: Open Neural Network Exchange, onnx, [2017-09-11], (原始内容存档于2017-09-09) 
  3. ^ Microsoft, Facebook Partner on AI Framework Interoperability. 
  4. ^ Model Server for Apache MXNet introduces ONNX support and Amazon CloudWatch integration. [8 February 2018]. (原始内容存档于2019-02-17). 
  5. ^ Codemod Toffee -> ONNX, toffee -> onnx. Change file names to match · pytorch/pytorch@6d8d5ba. GitHub. [2021-10-12]. (原始内容存档于2021-10-27) (英语). 
  6. ^ A model exporter for PyTorch by ezyang · Pull Request #2565 · pytorch/pytorch. GitHub. [2021-10-12]. (原始内容存档于2021-10-27) (英语).