news 2026/6/9 22:46:57

MindSpore 模型部署的2 种轻量落地方案

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
MindSpore 模型部署的2 种轻量落地方案

很多同学训练好 MindSpore 模型后,不知道怎么快速落地用起来,分享 2 个我常用的轻量部署方案,零基础也能搞定~

方案 1:端侧本地部署(用 MindSpore Lite)

适合把模型装到手机 / 嵌入式设备,步骤超简单:

先把训练好的模型转成 MindIR 格式(MindSpore 统一模型格式):

from mindspore import save_checkpoint, load_checkpoint, load_param_into_net, export from mindspore import Tensor, dtype as mstype # 加载训练好的模型 net = YourTrainedModel() param_dict = load_checkpoint("your_model.ckpt") load_param_into_net(net, param_dict) # 导出为MindIR input_tensor = Tensor([1,2,3], mstype.float32) # 模拟输入形状 export(net, input_tensor, file_name="model.mindir", file_format="MINDIR")

用 MindSpore Lite 转换为端侧模型:

去 MindSpore 官网下载 Lite 工具包,执行命令转格式:

mindspore-lite-converter --fmk=MINDIR --modelFile=model.mindir --outputFile=lite_model

3.端侧推理

from mindspore_lite import Model, Context # 初始化上下文 context = Context() context.target = ["cpu"] # 加载Lite模型 model = Model() model.build_from_file("lite_model.ms", context=context) # 推理 input_data = [Tensor([1,2,3], mstype.float32)] outputs = model.predict(input_data) print("推理结果:", outputs[0].asnumpy())

方案 2:Web 接口部署(用 Flask 封装)

适合把模型做成 HTTP 接口,供前端 / 其他服务调用:

安装依赖:

pip install flask mindspore

写接口代码(app.py):

from flask import Flask, request, jsonify from mindspore import load_checkpoint, load_param_into_net, Tensor, dtype as mstype import numpy as np app = Flask(__name__) # 加载模型(启动时只加载一次) net = YourTrainedModel() param_dict = load_checkpoint("your_model.ckpt") load_param_into_net(net, param_dict) @app.route("/predict", methods=["POST"]) def predict(): # 接收请求数据 data = request.json.get("input") input_tensor = Tensor(np.array(data), mstype.float32) # 模型推理 result = net(input_tensor).asnumpy().tolist() return jsonify({"result": result}) if __name__ == "__main__": app.run(host="0.0.0.0", port=5000)

测试接口:

用 Postman 或 curl 发 POST 请求:

curl -X POST http://127.0.0.1:5000/predict -H "Content-Type: application/json" -d '{"input": [1,2,3]}'

这两个方案我日常用得最多,不用复杂框架就能快速让模型 “跑起来”~

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/9 15:18:18

MindSpore自动混合精度训练中的梯度“消失”

一、问题现象:WGAN-GP在AMP训练中完全失效我们在MindSpore上复现WGAN-GP(带有梯度惩罚的Wasserstein GAN)模型。在FP32精度下,训练正常,判别器(Critic)损失能稳步下降,生成器&#x…

作者头像 李华
网站建设 2026/6/9 9:52:50

MindSpore 模型训练与推理全流程

第一章 概述 高效的模型训练与推理是 AI 应用落地的关键。昇腾硬件凭借其专为 AI 设计的架构,结合 MindSpore 框架的深度优化,能为开发者带来卓越的性能体验。本文将以一个典型的深度学习任务为例,详细阐述在昇腾硬件上基于 MindSpore 进行模…

作者头像 李华
网站建设 2026/6/9 16:25:28

SI2301S-ASEMI工业控制专用SI2301S

编辑:LLSI2301S-ASEMI工业控制专用SI2301S型号:SI2301S品牌:ASEMI沟道:PNP封装:SOT-23漏源电流:-2.3A漏源电压:-20VRDS(on):108mΩ批号:最新引脚数量:3封装尺寸&#xff…

作者头像 李华