摘要
本发明通过提供一种基于MXNet与ONNX的神经网络模型转换方法,解决了现有技术中不能进行MXNet框架下和ONNX框架下的神经网络模型相互转换的问题;该方法包括:通过MXNet模型加载及解析、算子对应与转换、转换数据处理、网络层转换、ONNX模型构建、ONNX模型保存等过程实现MXNet框架的神经网络模型向ONNX模型的转换过程;通过ONNX模型加载及解析、算子对应与转换、转换数据处理、计算图转换、MXNet模型导出等过程实现ONNX框架的神经网络模型向MXNet模型的转换过程;该方法实现了完成MXNet框架下的神经网络模型完整、高效地转换至ONNX框架,增强了神经网络模型的可移植性和灵活性。
技术关键词
神经网络模型
转换方法
节点
列表
框架
参数
符号
字典
对象
数据
标记
布局
元素
关系
系统为您推荐了相关专利信息
故障诊断方法
拓扑特征
机器学习分类模型
频域特征
多维度特征提取
容器单元
网关
流量隔离方法
多租户环境
管理插件