快速上手指南
前言
本文介绍客户如何快速使用SVP_NNN_PC发布包转换模型并在PC和SVP_NPU上进行推理。
与本文档相对应的产品版本如下。
说明:
本文如无特殊说明,Hi3519DV500与Hi3516DV500、Hi3516CV608与Hi3516CV610内容完全一致。
本文档主要适用于软件开发工程师。
掌握以下经验和技能可以更好地理解本文档:
- 熟悉Linux基本命令。
- 对机器学习、图像分析方法有一定的了解。
简介
功能框架

工具功能
SVP_NNN_PC组件包中,提供如下的工具链:
- AMCT:模型压缩工具(Advanced Model Compression Toolkit)对原始网络模型进行量化,量化是指对模型的权重(weight)和数据(activation)进行低比特处理,让最终生成的网络模型更加轻量化,从而达到节省网络模型存储空间、降低传输时延、提高计算效率,达到性能提升与优化的目标;
- Picovision:提供亲和SVP_NPU硬件的backbone或者算子实现。
- MindCmd:Python命令行工具,依赖CANN、AMCT(可选),支持网络模型端到端运行数据预处理、Ground Truth、AMCT、ATC、Simulate、板端推理、Dump、精度比对、性能分析,旨在提升网络模型移植、分析和优化的效率;
- Docker:提供docker环境
- PC Sample:提供sample示例
- Ascend-cann-toolkit:开发套件包,简称CANN。为开发者提供基于模式识别处理器SoC的相关算法开发工具包,旨在帮助开发者进行快速、高效的模型、算子和应用的开发。CANN主要提供如下工具:
- ATC:将开源框架网络模型通过ATC(Advanced Tensor Compiler)工具转换成适配图像分析引擎的离线模型;
- Simulate:提供功能、指令PC端模拟器,能够完整运行离线模型,加快网络模型的调试部署
- Acllib:开发板SVP___NPU提供的库
- Tools:提供Profiling性能分析工具和精度比对工具。Profiling性能分析工具用于采集和分析运行在SoC上的推理业务(应用或算子)各个运行阶段的关键性能指标,用户可根据输出的性能数据针对关键性能瓶颈做出优化以实现产品的极致性能;精度比对工具的定位是解决模型的精度问题,提供比对自有模型算子的运算结果与Caffe等标准算子的运算结果,以便确认误差发生的算子。
SDK组件:提供的C语言API库开发图像分析工具应用,用于实现目标识别、图像分类等功能。
开发环境准备
开发环境
该场景下的开发环境需要单独部署SVP_NNN_PC软件包,采用命令行的方式进行安装和使用。
简介
部署架构如图1所示。

板端环境:
板端环境中包含板端执行推理时需要的头文件、动态库、驱动ko以及sample。
开发环境
命令行方式开发环境
该场景下的开发环境需要单独部署CANN软件包,采用命令行的方式进行安装和使用。
使用流程
本章节介绍以原始训练好的模型如何在SVP_NPU上执行的整体使用流程,运行流程如图1所示。

下面以ONNX原始网络模型为例,说明运行流程:
当ONNX模型准备好后,可以直接使用ATC工具进行离线模型转换。
ATC离线模型转换后生成的om模型,可以在板端环境上使用ACL(Advanced Computing Language)接口做推理。
推理后如果遇到精度问题,可以选择Dump网络的中间层数据,并和ONNX的Dump结果做比较,来定位是哪一层的问题。
当推理性能不满足要求时,可以通过Profiling工具查看网络中每个算子的耗时以及带宽数据,可以通过分析瓶颈点,修改网络来提升整网性能。
说明:
上述流程中涉及的参考手册如下。
- AMCT模型压缩工具:
- 推荐使用AMCT的情况:
- 提高模型精度和减少量化误差,请参见《AMCT使用指南(Caffe)》“参数调优说明”章节和《AMCT使用指南(PyTorch)》“量化调优”章节。
- 量化误差分析,请参见《AMCT使用指南(PyTorch)》“量化误差分析”章节。
- 必须使用AMCT的情况:
- 4bit混合精度量化和高级离线量化,请参见《AMCT使用指南(PyTorch)》“自动混合精度量化”“高级离线量化”章节。
- 量化重训,请参见《AMCT使用指南(Caffe)》、《AMCT使用指南(PyTorch)》“量化感知训练”章节。
- 使用自定义量化算法,请参见《AMCT使用指南(PyTorch)》“自定义量化算法”章节。
- ATC工具模型转换:请参见《MindCmd使用指南》“模型转换”章节或《ATC工具使用指南》,将训练好的模型转换成平台识别的离线模型,应用程序指定模型资源后即可进行推理;具体支持的Caffe算子规格请参见《ATC工具使用指南》“算子规格说明”章节。
命令行开发环境安装
CANN(Compute Architecture for Neural Networks)是针对AI场景推出的异构计算架构,通过提供多层次的编程接口,支持用户快速构建AI应用和业务。本文档主要用于指导用户安装CANN开发环境,用于代码开发、编译等的开发活动(例如ATC模型转换、算子和推理应用程序的纯代码开发)。
安装流程如图1所示。

软件包获取
环境搭建前,请准备如下CANN软件包,用户根据具体板端环境,选择其中一个软件包进行安装。
表 1 软件包说明
其中_<version>_表示软件版本号。
环境要求
开发环境所要求的硬件及操作系统要满足以下条件。
表 1 环境信息
注:环境安装可具体参考《驱动和开发环境安装指南》文档。
软件包安装
软件包安装操作步骤:
如下命令中的*.run请替换为具体CANN软件包,CANN软件包路径:SVP_PC/SVP_NNN_PC_<version>/CANN/Ascend-cann-toolkit_<version>_linux.x86_64.run。
命令中所涉及的${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。例如,$HOME/Ascend/ascend-toolkit/<version>/x86_64-linux。
-
以CANN软件包的安装用户登录开发环境,切换到软件包所在路径。
-
增加安装用户对软件包的可执行权限。
在软件包所在路径执行ls -l命令检查安装用户是否有该文件的执行权限,若没有,请执行如下命令。
-
校验软件包。
执行如下命令,校验软件包安装文件的一致性和完整性。
-
执行如下命令进行安装(以下命令支持--install-path=<path>等参数,具体参数说明请参考文档《驱动和开发环境安装指南》)。
说明: - 如果以root用户安装,建议不要安装在非root用户目录下,否则存在被非root用户替换root用户文件以达到提权目的的安全风险。
- 如果要多个版本并存,请使用--install-path=<path>指定新版本的安装路径。
- 如果默认安装路径下已经安装了其他架构NPU的版本,请使用--install-path=<path>指定新的安装路径。否则安装和更新将覆盖原有版本。
若出现如下关键信息,则说明安装成功:
- 软件包默认安装路径:root用户/usr/local/Ascend;非root用户$HOME/Ascend
- 安装详细日志路径:${INSTALL_DIR}/ascend_install.log。
-
安装后软件包的安装路径、安装命令以及运行用户信息记录路径:${INSTALL_DIR}/<package_name>/ascend_install.info。
${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。例如,$HOME/Ascend/ascend-toolkit/<version>/x86_64-linux。
-
配置环境变量。
执行如下命令配置环境变量。
${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。例如,$HOME/Ascend/ascend-toolkit/<version>/x86_64-linux。
安装后处理
开发环境安装完毕,如果用户使用应用工程开发了相关程序,需要进行工程的编译和运行,用于生成相关二进制文件,进行工程编译之前,请先配置交叉编译环境。
根据开发环境安装时选择的CANN软件包的形态不同,交叉编译环境配置方法不同。
需要搭建交叉编译环境,具体方法请参考文档《驱动和开发环境安装指南》“交叉编译环境准备”章节。
手动部署
模型编译
模型编译
本章节介绍 ONNX 模型转换的基本操作, 使用 ATC工具将 ONNX 模型编译成 om 模型。请先参考“开发环境准备”章节完成开发环境搭建。本节示例模型为开源模型 yolov5s。
运行前准备
本章节所需模型yolov5s及相关依赖已在SVP_NNN_PC软件包中提供。
解压SVP_NNN_PC_<version>软件包提供的Sample目录下的samples.tar.gz
在当前解压目录下进入samples/2_object_detection/yolo目录
目录结构如下
├── build.sh //编译可执行文件脚本
├── caffe_model //caffe模型
│ ├── *.caffemodel
│ ├── *.prototxt
├── CMakeLists.txt //编译脚本,调用src目录下的CMakeLists文件
├── data
│ ├── ... //测试数据
├── inc
│ ├── model_process.h
│ ├── sample_process.h
│ └── utils.h
├── insert_op.cfg //aipp配置文件
├── model //atc 编译离线模型om存放路径
├── onnx_model //onnx模型
│ ├── *rpn.patch
│ ├── *.md
│ ├── *.onnx
├── README.md
├── script
│ ├── transferPic.py //将*.jpg转换为*.bin
│ ├── drawbox.py //画框脚本
├── src
│ ├── acl.json
│ ├── CMakeLists.txt
│ ├── main.cpp
│ ├── model_process.cpp
│ ├── sample_process.cpp
│ ├── utils.cpp
│ ├── *_rpn.txt
├── .project //工程信息文件,包含工程类型、工程描述、运行目标设备类型等
├── *.json //工具转换模型配置文件
该目录下,其中onnx_model文件夹存放原始的ONNX模型文件,insert_op.cfg配置了图像预处理的Data层使用图片文件格式, data文件夹用于存放生成量化参数时用的校准数据及测试数据,model文件夹用于存放编译生成的离线模型om。
命令说明
ATC工具详细的用法请参考《ATC工具使用指南》。
模型编译
以yolov5s.onnx为例, 在准备的环境中执行如下ATC命令编译生成离线模型。
atc --output="./model/yolov5" --insert_op_conf=./insert_op.cfg --framework=5 --save_original_model=true --model="./onnx_model/yolov5s.onnx" --image_list="images:./data/image_ref_list.txt"
备注:如果操作产品为Hi3516CV610/Hi3516CV608,需额外配置参数--soc_version=Hi3516CV610/--soc_version=Hi3516CV608。
若提示如下信息,则说明模型转换成功
成功执行命令后,在model目录下可查看离线模型(yolov5_original.om)。
模型部署运行
编译示例
环境准备
- Cmake版本大于等于 X.X.X(X.X.X表示版本号3.10.2(Hi3519DV500)或版本号3.22.1(Hi3516CV610))
-
设置环境变量
必选环境变量(如下环境变量中${install_path}以软件包使用默认安装路径为例进行说明)
执行如下命令配置环境变量。
-
本地已安装开发板所需的交叉编译链工具链
Hi3519DV500/Hi3516DV500:
- 切换到/opt/linux/x86-arm/aarch64-v01c01-linux-musl-gcc/bin路径查看是否存在aarch64-v01c01-linux-musl-g++交叉编译器。
- 切换到/opt/linux/x86-arm/aarch64-v01c01-linux-gnu-gcc/bin路径查看是否存在aarch64-v01c01-linux-gnu-g++交叉编译器。
Hi3516CV610/Hi3516CV608:
切换到/opt/linux/x86-arm/arm-v01c02-linux-musleabi-gcc/bin/路径查看是否存在arm-v01c02-linux-musleabi-g++交叉编译器。
本地编译
说明:
本文对应芯片工具链如下:
- Hi3519DV500/Hi3516DV500:aarch64-v01c01-linux-gnu-gcc/aarch64-v01c01-linux-musl-gcc
- Hi3516CV610/Hi3516CV608:arm-v01c02-linux-musleabi-gcc
切换到样例目录2_object_detection/yolo,直接执行命令:
同时在out目录下生成功能仿真可执行文件(func_main),指令仿真可执行文件(inst_main)和板端可执行文件(board_glibc_main,board_musl_main)。其中:
- board_glibc_main由编译链aarch64-v01c01-linux-gnu-gcc编译的可执行文件;
- board_musl_main由编译链aarch64-v01c01-linux-musl-gcc/arm-v01c02-linux-musleabi-gcc 编译的可执行文件。
运行示例
该场景下的开发环境需要单独部署CANN软件包,采用命令行的方式进行安装和使用。
数据准备
-
准备离线模型om
确认model目录下有yolov5_original.om离线模型。模型编译,请参见“模型编译”。
-
准备输入图片
切换到“2_object_detection/yolo/data”目录下,执行transferPic.py脚本,将*.jpg转换为*.bin,同时将图片分辨率缩放为模型输入所需的640*640。在“样例目录/data”目录下生成*.bin文件。
须知: - 如果执行脚本报错“ModuleNotFoundError: No module named 'PIL'”,则表示缺少Pillow库,请使用**pip3.7.5 install Pillow --user**命令安装Pillow库。
- “5”表示生成yolov5
仿真运行
切换到可执行文件main所在2_object_detection/yolo/out目录,给可执行文件main加执行权限
运行可执行文件
功能仿运行:
指令仿运行:
若提示如下信息,则说明运行成功
“5”表示生成yolov5
开发板运行
环境安装
- 板端环境安装请参见《Hi35xxVxxx SDK 安装以及升级使用说明》。
- SVP ACL接口使用说明请参见《应用开发指南》中“SVP ACL API参考”章节。
- 需要将SVP_NPU相关库路径加入到系统环境变量LD_LIBRARY_PATH(如Hi3519DV500,库路径为/xxx(客户自定义)/smp/a55_linux/mpp/out/lib,其他解决方案类似参考)。
- 板端开发所依赖的文件清单如表1所示。
表 1 板端开发依赖文件
|
libss_mpi_km.so (Hi3516CV610不依赖这个库) libprotobuf-c.so -> libprotobuf-c.so.1.0.0 libprotobuf-c.so.1 -> libprotobuf-c.so.1.0.0libprotobuf-c.so.1.0.0 |
|
运行准备
运行准备:
使用mount命令将NFS服务器上的目录2_object_detection/yolo挂载到板端环境的指定目录。
使用mount命令将NFS服务器上的SVP_NNN_PC_<version>软件包提供的Sample目录下的opencv依赖库目录,即samples/third_party/<gcc_version>/opencv/lib,挂载到板端环境指定目录$ BOARD_OPCV_LIB。
<gcc_version>:根据开发板编译链选择。
- 开发板编译工具链为aarch64-v01c01-linux-musl-gcc或arm-v01c02-linux-musleabi-gcc, 则<gcc_version>为musl;
- 开发板编译工具链为aarch64-v01c01-linux-gnu-gcc, 则<gcc_version>为glibc。
挂载开发板:
-
以root用户登录Linux服务器(Ubuntu操作系统),安装NFS服务并配置共享目录。
确保环境连网且源可用的前提下,安装NFS服务,可参考如下命令安装,如果安装过程中提示已安装NFS服务,则无需重复安装:
在“/etc/exports”文件中配置共享目录,配置完成后,执行/etc/init.d/nfs-kernel-server restart命令重启NFS服务使配置生效。可参考如下配置段(添加在文件的末尾),斜体部分请根据实际情况替换,服务器绝对路径表示Linux服务器的共享目录,如不存在,请提前创建。
-
命令示例如下,其中,path表示板端环境上的目录,需根据实际情况修改。
运行
在开发板上设置开发板运行所需库文件路径
切换到开发板可执行文件main所在yolo/out目录,给可执行文件main加执行权限
运行可执行文件
说明:
- “5”表示生成yolov5。
- 根据板端环境选择可执行文件(board_glibc_main或board_musl_main)运行。
运行后处理
模型执行后,基于推理结果,输出检测结果并保存至out目录下xxx_detResult.txt中,其中第一行为输入图片的W和H的大小,下面每一行有6个值,分别是对应框的classId, score, leftX, leftY, rightX, rightY。
执行结果保存为out_img_yolov5.jpg,如图1所示。

自动部署
本节示例为使用MindCmd工具完成开源模型yolov5s.onnx的一键式部署。
MindCmd工具准备
MindCmd是一个Python命令行工具,依赖Python3.7.5(Hi3516CV610形态下为Python3.10.12)和CANN。如果还没有安装CANN软件包,请先参考软件包安装章节完成安装。
请参考如下步骤安装并配置MindCmd工具:
-
安装MindCmd
以CANN软件包的安装用户登录开发环境。MindCmd工具安装包位于SVP_PC/SVP_NNN_PC软件包中,进入安装包所在路径,并执行如下命令安装:
若出现如下关键信息,则说明安装成功:
Successfully installed mindcmd-x.x.x
-
执行如下命令,查看MindCmd工具默认配置的CANN软件包安装路径:

如果不满足,请参考如下命令重置CANN软件包的实际安装路径:
说明: 命令中的“/usr/local/Ascend/ascend-toolkit/svp_latest”为执行示例,请根据实际安装路径进行替换。
-
查看版本信息
执行如下命令,查看MindCmd工具的版本和工具在步骤2中配置的CANN软件包版本信息:
如果返回以下关键信息(版本信息以实际安装为准),说明MindCmd工具安装成功。

说明: 图片示例返回的MindCmd工具和CANN软件包版本信息仅供参考,会根据用户环境实际安装的工具版本打印。
资源准备
本示例所需的模型和相关资源文件均位于SVP_NNN_PC软件包中的2_object_detection/yolo样例目录,请先执行如下命令解压:
仿真运行
切换到样例目录2_object_detection/yolo,并执行如下命令启动仿真运行:
cd samples/2_object_detection/yolo/
mindcmd oneclick onnx --model ./onnx_model/yolov5s.onnx --image_list ./data/image_ref_list.txt --aipp ./insert_op.cfg --rpndata ./src/yolov5_rpn.txt
若出现如下关键信息,则说明功能仿真运行成功:

开发板运行
请参考《Hi35xxVxxx SDK 安装以及升级使用说明》完成板端基础环境安装。参考《驱动和开发环境安装指南》“OpenSSH服务搭建”章节完成板端ssh环境安装,服务器NFS环境安装。
以上环境安装完成后,请参考如下步骤在开发板运行yolov5s.onnx:
-
在样例目录2_object_detection/yolo下创建ssh.cfg文件,文件格式和详细内容示例如下:
[ssh_config] # board ip BOARD_IP=x.x.x.x # board work directory, mount to $HOST_MOUNT_PATH BOARD_MOUNT_PATH=/home/MindCmdUser/board_workspace/ # local host ip HOST_IP=x.x.x.x # to avoid bottlenecks caused by copying test resources, store test resources in this path as much as possible. HOST_MOUNT_PATH=${SHARE_DIR}/host_workspace # board user name USER=${username} # board user's password PASSWORD=${password} # default port is 22 PORT=22
说明: - MindCmd工具上板推理时会根据以上配置文件提供的信息,自动执行mount、umount操作。
- HOST_MOUNT_PATH配置的路径不能超出服务器所配置的NFS共享目录范围,否则可能会导致mount失败。
- 示例中的${SHARE_DIR}、${username}、${password}请根据实际情况替换。
-
执行如下命令设置MindCmd工具工作路径:
说明: - MindCmd工具与板端采用mount方式共享资源,因此上板推理时MindCmd工具的工作路径不能超出步骤1配置的服务器侧挂载路径(HOST_MOUNT_PATH)范围。
- 示例中的${SHARE_DIR}请根据实际情况替换。
-
执行如下命令打开MindCmd工具开发板运行开关:
-
执行如下命令启动开发板运行:
cd samples/2_object_detection/yolo/ mindcmd oneclick --ssh_config ./ssh.cfg onnx --model ./onnx_model/yolov5s.onnx --image_list ./data/image_ref_list.txt --aipp ./insert_op.cfg --rpndata ./src/yolov5_rpn.txt若出现如下关键信息,则说明开发板运行成功:

进阶指南
精度分析
本章节主要介绍常见精度问题的定位和分析。精度问题也叫精度掉点,是指在相同的输入下,离线模型的推理结果和原始模型的推理结果偏差超出业务可接受范围。
常见的精度问题来源
- 数据预处理:离线模型与原始模型的预处理不同引入。
- 模型量化:AMCT量化或ATC量化时引入。
- 指令化阶段:ATC转换离线模型时引入。
- 模型后处理:离线模型与原始模型后处理方式不同引入。
界定精度问题的引入阶段
- 使用浮点框架的输入数据和推理工程的输入数据做比对,如果掉点较多说明数据预处理过程引入了问题。
- 使用相同的输入做推理,分别Dump原始模型、量化模型和离线模型的逐层输出。
- 对比原始模型和Fake Quant模型的Dump结果,如果掉点较多说明量化过程引入了问题。
- 对比Fake Quant模型和离线模型的Dump结果,如果掉点较多说明指令化过程引入了问题。
使用MindCmd一键推理进行精度比对
本节示例为MindCmd工具准备完成sample网络yolov5s_cpu.onnx的一键精度比对。所需模型文件、推理数据均位于SVP_NNN_PC软件包中的2_object_detection/yolo样例目录,请参考资源准备章节完成资源准备。
执行如下命令进行一键精度比对:
cd samples/2_object_detection/yolo
mindcmd config --global oneclick_switch.is_npu_run=0
mindcmd oneclick onnx --model ./onnx_model/yolov5s_cpu.onnx --image_list ./data/image_ref_list.txt --aipp ./insert_op.cfg
如图1所示,MindCmd一键推理会自动执行原始模型推理、量化模型推理、功能仿真推理,并将推理结果进行精度比对。

精度调优建议
- 使用MindCmd一键推理,定位数据预处理、模型压缩和模型后处理引入的精度问题。
- 若浮点框架的输入数据和推理程序的输入数据存在精度问题,则请排查推理数据是否使用了相同的预处理方式。
- 若原始模型与ATC量化模型的推理结果存在精度问题时,请修改mindcmd.ini使能AMCT量化,和Fake Quant比较。(AMCT仅支持PyTorch和Caffe模型)。
- 若原始模型与AMCT量化模型的推理结果存在精度问题时,请参考《AMCT使用指南》进行精度调优,或使用QAT解决
- 若指令编译阶段引入的精度问题,请对比cnn_net_tree.dot和原始模型(*.dot查看请参考《ATC工具使用指南》“输出文件说明”章节),查看是否修改了网络结构,比较被修改网络结构层的后面层的相似度。
若按照以上建议排查问题未发现根因,可以提单进一步分析,需要提供能复现问题的原始模型、Fake Quant模型、om模型、推理文件、校准集、ATC命令(包含数据预处理配置文件)、转换log打印等。
性能分析
本章节介绍如何通过MindCmd进行快速获取帧率、带宽、内存、耗时等性能数据,分析性能问题并进行性能调优。
使用MindCmd工具获取性能数据
安装好MindCmd后,通过修改全局配置,开启上板Profiling来获取性能数据。
-
修改配置项,示例:
-
参考配置ssh.cfg,新建文件_$HOME_/ssh.cfg。
说明: - 请参考《MindCmd使用指南》“NFS环境搭建”章节配置NFS环境。
- MindCmd工作路径与板端挂载路径的配置请参考《MindCmd使用指南》“关于工作路径、挂载路径、数据卷路径和NFS共享路径”章节。
-
执行Oneclick
提前准备模型文件、图片集等,然后使用MindCmd进行一键推理。
以yolov5s.onnx为例
cd SVP_NNN_PC_<version>/Sample/samples/2_object_detection/yolo/onnx_model/ mindcmd oneclick onnx -m yolov5s.onnx -i ../data/image_ref_list.txt -r ../src/yolov5_rpn.txt执行Oneclick后,MindCmd会将采集到的性能数据保存在csv表格中(如图1所示),csv表格存放位置如图2所示,模型的整网性能数据打印如图3所示。



通过得到的Profiling数据,可以了解模型在推理过程中各层耗时及各层耗时占比、带宽、帧率等性能数据(如图4所示),根据获取的性能数据,我们能分析出是否存在关键性能瓶颈,辅助模型性能调优。
图 4 AI Core数据表(op_summary_*.csv)

Profiling采集的数据如下:
-
硬件的性能数据,包括:AI Core、AI Vector Core、AI CPU的系统硬件性能指标。
-
软件的性能数据,统计ACL接口的调用时间。
可以通过在表格内对某一项数据进行降序排序,从而对耗时长、占用内存大的层进行分析,判断是否存在性能瓶颈。
-
如果想了解Profiling各项数据的详细说明,请参考《MindCmd使用指南》“性能分析数据说明”章节。
性能问题分析及性能调优
-
针对性能结果数据分析原因
根据性能数据结果进行分析,可以比较得出耗时较长、占用内存和带宽较多的接口和算子,针对这些接口和算子做进一步的剖析,从而定位到性能问题根因。
优化建议参考:
-
当整体耗时,整网Mac Ppen Ratio超过60%,基本不存在性能瓶颈。
-
如果是CUBE层,当Mac Ppen Ratio超过70%,基本不存在性能瓶颈。
-
如果是非CUBE层,当Vec Ppen Ratio超过50%且不为0,基本不存在性能瓶颈。
-
如果上述场景均不符合,则分析DLD Ratio,WLD Ratio,Dstr Ratio的数据
- 如果DLD Ratio高于70%,则可能DLD阻塞,建议裁剪输入数据量,如减通道、减少数据搬移或数据转换层如slice,concat,reshape,permute等。
- 如果WLD Ratio高于80%,则可能是WLD阻塞,建议减少权重数据量,如对该层的权重做4bit量化。
- 如果Dstr Ratio高于70%,则可能是Dstr阻塞,建议裁剪feature map的结果大小。
- 如果Mac Ppen Ratio、Vec Ppen Ratio、DLD Ratio、WLD Ratio、Dstr Ratio都低于50%,建议提单进行反馈。
-
-
使用AMCT对模型进行进一步量化压缩
如果希望进一步提升性能,还可以参考《AMCT使用指南》使用AMCT工具进行混合精度量化调优,然后重新对精度进行分析。
-
设计高性能算子模型
通过修改网络连接方式和设置算子属性,可对网络模型的性能做进一步优化。详见《ATC工具使用指南》“性能优化推荐设置”章节。
高阶指南
自定义算子
- ATC提供的接口开发自定义算子,用户可开发自定义算子以提高网络运行效率。
- 自定义算子更详细的用法请参考《ATC自定义算子开发指南》手册。
IR构图
- ATC提供Graph接口构建网络模型,并将其转换成图像分析引擎支持的离线模型,模型转换过程中可以实现算子调度的优化、权值数据重排、内存使用优化等,可以脱离设备完成模型的预处理。
- Graph更详细的用法请参考《ATC Graph开发指南》手册。
AMCT Pytorch 4bit量化
AMCT Pytorch支持选择多个压缩率,做多卡多任务的混精aptq搜索。
AMCT工具安装包位于SVP_PC/SVP_NNN_PC软件包中,请先安装amct_pytorch包,参考命令如下:
cd SVP_NNN_PC_<version>/amct/amct_pytorch/
pip install hotwheels_amct_pytorch-<version>-none-linux_x86_64.tar.gz
以sample中提供的yolov5为例:
-
解压sample包
-
进入yolov5目录
-
参考SVP_NNN_PC_<version>/amct/amct_pytorch/sample/yolov5/README.md 中的“1.下载对应的代码、权重和数据集”和“2.导入量化补丁”进行操作,提前准备好资源文件。
-
执行多任务混精搜索脚本,命令参数介绍如表1所示。
python -m torch.distributed.run --master_port 49252 --nproc_per_node 5 yolov5s_mix_aptq_search.py --device 0,1,2,3,4 --mix_precision_search_configs ./configs/mix_precision_search_configs.yaml表 1 命令参数说明
混精压缩率配置文件,sample的mix_precision_search_configs.yaml中提供了5档压缩率,可根据需要自行调整。
-
yolov5s_mix_aptq_search.py会保存每一档压缩率的混精aptq输出结果,包括量化模型、权重、量化配置等。
以compress_1.7为例,混精aptq输出结果保存在 ./work_dir/mix_aptq_search/yolov5s/compress_1.7/quant_output/ 下,如图1所示。

-
每一档压缩率的精度保存在 ./work_dir/mix_aptq_search/yolov5s/search_result.csv,如图2所示,如果当前压缩率的精度已满足要求,可以选择对应压缩率执行后的 yolov5s_quant_param_record.txt 和 yolov5s_deploy_model.onnx 进行部署。

-
如果步骤6中没有符合精度要求的压缩率,可以选择某档精度较优的压缩率做qat重训。
以compress_1.7为例:
python -m torch.distributed.run --master_port 49252 --nproc_per_node 8 train_a8w4_qat.py --device 0,1,2,3,4,5,6,7 --cos-lr --batch-size 256 --data ./yolov5-7.0/data/coco.yaml --cfg ./yolov5-7.0/models/yolov5s.yaml --hyp ./yolov5-7.0/data/hyps/hyp.scratch-yolo5n-qat.yaml --weights ./yolov5-7.0/yolov5s.pt --epochs 300 --sync-bn --cos-lr --quant --ao --config_def ./work_dir/mix_aptq_search/yolov5s/compress_1.7/quant_output/mixed_quantize.yml如果使用其他的压缩率,请替换--config_def参数值为其他档位压缩率的混精配置文件。
-
如果重训后精度仍不满足要求,可调整 ./yolov5-7.0/data/hyps/hyp.scratch-yolo5n-qat.yaml 中参数配置,重新执行步骤7。
注意:
配置的显卡数量需要与混精压缩率配置文件中压缩率的档数一致。
更多量化接口的详细介绍,请参考《AMCT使用指南(Pytorch)》“5 接口说明”章节。
安全方案
- 安全方案是指SVP_NPU安全,为基于OP-TEE开发、部署保护模型和数据安全。
- 安全方案更详细的用法请参考《安全开发指南》手册。
须知:
Hi3516CV610由于硬件限制不支持安全方案。
附录
参考文档
表 1 参考文档