昇思 ONNX 迁移流程：双向转换实践

昇思MindSpore

6人浏览 · 2026-08-27 15:56:56

昇思MindSpore · 2026-08-27 15:56:56 发布

昇思 MindSpore 对 ONNX 提供双向迁移能力：既支持将 MindSpore 模型导出为 ONNX 格式用于跨框架部署，也支持将第三方 ONNX 模型（如 PyTorch/TensorFlow）迁移至 MindSpore，依托MindConverter、export 接口、ONNX Runtime 校验构建全链路工具链，保障转换精度与性能，是大模型跨框架迁移、昇腾部署的核心流程。

一、迁移总览：双向打通 ONNX 生态

1. MindSpore → ONNX（导出部署）

核心能力：训练后 ckpt / 模型直接导出 ONNX，支持静态 / 动态输入、多输入输出、混合精度，适配昇腾、GPU、CPU 多硬件。
典型场景：模型跨框架部署（如 PyTorch 推理）、端侧设备（M3562）部署、第三方推理工具集成。

2. ONNX → MindSpore（迁移适配）

核心能力：通过 MindConverter 将 ONNX 图结构自动转为 MindSpore 脚本 + ckpt 权重，支持算子映射、权重对齐、转换报告生成。
典型场景：PyTorch/TensorFlow 模型迁移至昇腾训练、基于 MindSpore 生态二次开发、国产硬件适配。

二、流程一：MindSpore 导出 ONNX（代码实践）

1. 环境准备

pip install mindspore onnx  # 安装依赖

2. 导出代码（静态输入，推荐）

import mindspore as ms
from mindspore import Tensor, export, load_checkpoint, load_param_into_net
import numpy as np

# 1. 定义模型（以ResNet50为例）
class ResNet50(ms.nn.Cell):
    def __init__(self):
        super().__init__()
        # 模型层定义（省略细节）
        self.conv1=ms.nn.Conv2d(3, 64, kernel_size=7, stride=2)
    def construct(self, x):
        x=self.conv1(x)
        return x

# 2. 加载权重并初始化模型
net=ResNet50()
param_dict=load_checkpoint("resnet50.ckpt")  # 训练好的权重
load_param_into_net(net, param_dict)
net.set_train(False)  # 推理模式

# 3. 构造输入并导出ONNX
input_tensor=Tensor(np.random.randn(1, 3, 224, 224).astype(np.float32))
export(net, input_tensor, file_name="resnet50", file_format="ONNX")
print("ONNX导出完成：resnet50.onnx")

3. 动态输入导出（适配可变尺寸）

# 动态轴设置（batch、height、width可变）
export(
    net,
    input_tensor,
    file_name="resnet50_dynamic",
    file_format="ONNX",
    dynamic_axes={"input": [0, 2, 3], "output": [0]}  # 动态维度
)

三、流程二：ONNX 迁移至 MindSpore（代码 + 工具）

1. 准备 ONNX 模型（以 PyTorch 导出为例）

# PyTorch导出ONNX（示例）
import torch
model=torch.hub.load('pytorch/vision:v0.10.0', 'resnet50', pretrained=True)
model.eval()
dummy_input=torch.randn(1, 3, 224, 224)
torch.onnx.export(model, dummy_input, "resnet50.onnx", opset_version=11)

2. 用 MindConverter 迁移（命令行，核心）

# 安装工具
pip install mindinsight

# 迁移命令（关键参数：模型路径、输入尺寸、输入/输出节点）
mindconverter \
  --model_file ./resnet50.onnx \
  --shape 1,3,224,224 \
  --input_nodes input \
  --output_nodes output \
  --output ./converted_model \
  --report ./converted_model/report

输出内容：model.py（MindSpore 脚本）、model.ckpt（权重）、转换报告、权重映射表。

3. 加载验证迁移模型

# 导入迁移后的模型
from converted_model.model import ResNet50
net=ResNet50()
load_param_into_net(net, load_checkpoint("./converted_model/model.ckpt"))

# 推理验证
input_data=Tensor(np.random.randn(1, 3, 224, 224).astype(np.float32))
output=net(input_data)
print("迁移模型推理完成，输出shape：", output.shape)

四、精度校验：确保迁移一致性（关键步骤）

import onnxruntime as ort

# 1. ONNX模型推理
session=ort.InferenceSession("resnet50.onnx")
onnx_input={"input": np.random.randn(1, 3, 224, 224).astype(np.float32)}
onnx_output=session.run(None, onnx_input)[0]

# 2. MindSpore模型推理
ms_output=net(Tensor(onnx_input["input"])).asnumpy()

# 3. 精度校验（误差<1e-5为合格）
assert np.allclose(onnx_output, ms_output, atol=1e-5), "精度不一致"
print("ONNX与MindSpore模型精度一致，迁移成功！")