news 2026/7/7 10:51:24

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/7/7 6:18:02

CentOS7 设置 RAID5

1.添加磁盘 在vm虚拟机中添加4个磁盘来组成RAID5,这里我就添加4个2GB的磁盘,因为我不是真的使用: 2.硬盘初始化 我们首先对于磁盘就行初始化,我们先运行 lsblk看一下磁盘情况 我们现在开始初始化,那么我们需要用到的是…

作者头像 李华
网站建设 2026/7/7 18:00:03

Sidebar Diagnostics:Windows桌面硬件监控终极解决方案

Sidebar Diagnostics:Windows桌面硬件监控终极解决方案 【免费下载链接】SidebarDiagnostics A simple sidebar for Windows desktop that displays hardware diagnostic information. 项目地址: https://gitcode.com/gh_mirrors/si/SidebarDiagnostics Side…

作者头像 李华
网站建设 2026/7/7 15:44:08

ChatMCP跨平台AI助手全面使用指南

ChatMCP跨平台AI助手全面使用指南 【免费下载链接】chatmcp ChatMCP is an AI chat client implementing the Model Context Protocol (MCP). 项目地址: https://gitcode.com/gh_mirrors/ch/chatmcp 项目概述 ChatMCP是一个基于Model Context Protocol的创新型AI聊天客…

作者头像 李华
网站建设 2026/7/7 16:21:11

sentry 命令_前端异常监控之 Sentry的部署和使用

由于最近在整理前端异常监控相关内容,所以自己在虚拟机搭建部署了一下Sentry,把搭建过程及一些自己踩得坑整理如下 一、Sentry部署 Sentry搭建有两种方式: 通过Python安装 教程地址 通过Docker容器安装 教程地址 我本地是用Docker进行搭…

作者头像 李华
网站建设 2026/7/7 18:32:34

突破性工具使用框架:重构LLM多跳问答新范式

突破性工具使用框架:重构LLM多跳问答新范式 【免费下载链接】tinker-cookbook Post-training with Tinker 项目地址: https://gitcode.com/GitHub_Trending/ti/tinker-cookbook 如何让大型语言模型自主制定搜索策略,高效完成复杂事实查询&#xf…

作者头像 李华