当前位置: 首页 > news >正文

模型部署笔记

目录

  • 模型部署工作
  • ONNX存在的意义
  • ONNX(Open Neural Network Exchange)
    • ONNX示例
    • 模型推理示例
    • Batch调整
    • 量化
    • 量化方式
  • 常见问题

模型部署工作

  1. 训练好的模型在特定软硬件平台下推理
  2. 针对硬件优化和加速的推理代码

训练设备平台:
CPU、GPU、DSP

ONNX存在的意义

模型与硬件之间的对应关系适配复杂度mxn的结果,导致开发复杂化、效率较低等问题

在这里插入图片描述
使用一种模型表达结构将训练框架的输出结构统一化,将模型部署复杂度从mxn转变为m+n

在这里插入图片描述
ONNX变身是一种模型格式,属于文本,不是程序,无法直接在设备上运行。因此,需要软件栈去加载ONNX模型,使其在硬件设备上高效推理。这个软件栈指的是模型的推理框架。推理框架分类硬件商自研和通用推理框架。自研推理框架底层优化较好,推理的计算效率较高,不具备普适性,无法应用到其他的芯片上。而通用推理框架是具备通用性,可应用在不同的软硬件平台下,降低开发难度,提升开发的效率,不需要用户关注底层框架,只需要将接口对应完成即可。
在这里插入图片描述整体推理过程:先使用模型框架训练完成模型后,转换为ONNX模型结构,在使用推理框架,将ONNX模型高效地运行在软硬件平台下。
在这里插入图片描述

ONNX(Open Neural Network Exchange)

一种针对机器学习所设计的开放式的文件格式,用于存储训练好的模型。不同的训练框架可采用相同格式存储模型并交互。由微软,亚马逊,Facebook和IBM等公司共同发起。
在这里插入图片描述

ONNX示例

使用torch.onnx.export进行onnx模型导出。
在这里插入图片描述
导出的onnx模型之后,进入netron.app进行模型结构可视化操作

在这里插入图片描述
ResNet的ONNX模型导出
在这里插入图片描述
参数讲解
在这里插入图片描述

模型推理示例

使用mmdeploy中示例,应用ONNX Runtime进行模型推理
在这里插入图片描述
使用TensorRT对ONNX模型进行推理
在这里插入图片描述

Batch调整

上述生成ONNX时,是使用(1, X,X,X)图像输入是1维度,导致模型推理时也为一张图一张图的推理,效率较慢,使用多张图进行模型推理,使用dynamic_axes参数即可实现。

在这里插入图片描述
在这里插入图片描述
在TensorRT模型数据格式转换中进行修改,minShapes和maxShapes参数设置
在这里插入图片描述

量化

除了使用Batch增加模型推理速度外,还可以使用量化进行加速。一般而言,使用float16不会影响模型的精度变化,而使用int8存储格式后模型精度会略有下降。
在这里插入图片描述

量化方式

  1. 训练后量化:Post-training quantization(PTQ)
  2. 训练时量化:Quantization-aware training(QAT)

如果使用的是QAT的话,是在模型训练过程中已经实现了模型精度的转换,使用onnx数据结构转换即可,而如果使用的是PTQ的话,需要在对应的计算平台上进行精度转换。

对称量化:取模型的数值对称区间,对应到INT8(-127,127)中,可以得到一个量化系数(简单理解为比例系数,127/6),从而将FP32中的数值与INT8进行对应。
在这里插入图片描述
在TensorRT中被称为calibration
在这里插入图片描述

常见问题

在这里插入图片描述

http://www.lryc.cn/news/22080.html

相关文章:

  • 多线程之wait和notify
  • MVCC 当前读 快照读 RC read view RR下事务更新不会丢失
  • NCRE计算机等级考试Python真题(二)
  • 借助IBM Spectrum LSF为芯片行业大幅提升算力,预测未来
  • 力扣-换座位
  • DFT基本入门介绍
  • 做「增长」必须懂的6大关键指标
  • Linux:soft lockup 检测机制
  • 天线理论知识4——非频变天线
  • 基础架构组件选型及服务化
  • leetcode-每日一题-1247(中等,数学逻辑)
  • 前端面试题 —— 计算机网络(一)
  • 分布式-分布式缓存笔记
  • 【反序列化漏洞-01】为什么要序列化
  • 用c语言模拟实现常用字符串函数
  • 在 Flutter 中使用 webview_flutter 4.0 | 基础用法与事件处理
  • JavaWeb--Servlet
  • Linux启动过程
  • 面试资料整理——C++
  • 【ArcGIS Pro二次开发】(9):GeoProcessing工具和自定义工具的调用
  • 皕杰报表斜线单元格、图表里或导出pdf的中文显示小方块解决方案
  • python读写hdfs文件的实用解决方案
  • RK3399+FPGA+MIPI 方案细节之subLVDS to MIPI处理
  • Vue组件是怎样挂载的
  • gcc: 编译选项:-fdelete-null-pointer-checks、-fno-delete-null-pointer-checks
  • 周赛334(前缀和、贪心+双指针、Dijkstra求最短路径、二分答案)
  • imx6ull——I2C驱动
  • Spring Cache的基本使用与分析
  • 【安全知识】——端口复用隐藏后门
  • Tina_Linux量产测试使用指南_new