小米MACE正式支持ONNX模型,AI开发者从训练到部署更加无缝!
ONNX(Open Neural Network Exchange)是微软、亚马逊、Facebook 和 IBM 等公司共同开发的一种为存储深度学习模型设计的开放式文件格式。它使得不同的框架(Pytorch,MXNet)的模型之间可以相互转换。目前 ONNX 官方支持的深度学习框架有 Caffe2、Microsoft Cognitive Toolkit、MXNet 和 PyTorch,并且 TensorFlow 和 CoreML 也非官方的支持 ONNX。
小米 MACE 首次支持 ONNX 模型,受众范围为习惯 PyTorch、Caffe2、MXNet 框架的开发者,他们可轻松通过 MACE 转化至移动终端。MACE 加入 ONNX 家庭后,不仅带来大批可观优秀的中国开发者,更能通过模型的格式统一拉动芯片横向评测的绝对公正化。让人兴奋的是,伴随 5G 将至,边缘计算需求协同云激增,MACE 子计划 MODEL ZOO 也将被 ONNX 模型不断扩充,反哺开发者。
MACE 开源地址: https://github.com/XiaoMi/mace