2024云边一体化人工智能平台模型格式转换标准.pdf

云边一体化人工智能平台模型格式转换标

准

目  次

前言II

1范围1

2规范性引用文件1

3术语、定义和缩略语1

3.1术语和定义1

3.2缩略语1

4云边一体化人工智能平台架构与模型格式转换框架的关系2

5云边一体化人工智能平台模型格式转换框架3

6不同深度学习框架下模型导出格式标准3

6.1Tensorflow框架的模型格式3

6.2PyTorch框架的模型格式5

6.3Caffe框架的模型格式6

6.4MXNet框架的模型格式6

6.5XGBoost框架的模型格式6

6.6PaddlePaddle框架的模型格式7

7不同深度学习框架下模型格式转换标准7

7.1Tensorflow与ONNX的相互转换转ONNX8

7.2PyTorch模型转ONNX9

7.3MXNet模型转ONNX9

7.4Keras模型转ONNX9

7.5Caffe2模型转ONNX10

7.6Scikit-Learn模型转ONNX10

附录A（规范性附录）规范性附录标题12

附录B（资料性附录）资料性附录标题13

I

云边一体化人工智能平台模型格式转换标准

1范围

本标准规定了云边一体化人工智能平台兼容各种主流深度学习框架训练后的模型格式以及不同模

型之间的转换格式。

本标准适用于在同一个人工智能平台下，可以满足不同业务常见的模型格式需求，方便进行统一的

管理和使用。

2规范性引用文件

下列文件对于本文件的应用是必不可少的。凡是注日期的引用文件，仅注日期的版本适用于本文件。

凡是不注日期的引用文件，其必威体育精装版版本（包括所有的修改单）适用于本文件。

GB/T5271.28-2001人工智能基本概念与专家系统

GB/T37739-2019信息技术云计算平台即服务部署要求

3术语、定义和缩略语

3.1术语和定义

下列术语和定义适用于本文件。

模型转换Modeltransformation

将某个抽象层次或形式的模型转换为另一个抽象层次或形式的模型。

3.2缩略语

下列缩略语适用于本文件。

APIApplicationProgrammingInterface应用程序编程接口

AIArtificialIntelligence人工智能

SDKSoftwareDevelopmentKit软件开发工具包

MLMachineLearning机器学习

1

DLDeepLearining深度学习

4云边一体化人工智能平台架构与模型格式转换框架的关系

面向人工智能领域的公共服务平台以京东AI开放平台为基础，以云边一体化的场景应用为导向，

面向各行业提供统一的数据接入、模型开发、模型训练、模型评估和服务发布的一站式流程。其中模型

格式转换框架是在云边一体化人工智能平台的模型开发环境中进行提取，形成在各种深度学习框架下模

公

开

型格式转换的API，如图1所示：

2

0

2

2

年

1

2