AI开发平台ModelArtsAI开发平台ModelArts

计算
弹性云服务器 ECS
云耀云服务器 HECS
裸金属服务器 BMS
弹性伸缩 AS
镜像服务 IMS
专属主机 DeH
函数工作流 FunctionGraph
云手机 CPH
VR云渲游平台 CVR
特惠算力专区
存储
对象存储服务 OBS
云硬盘 EVS
云备份 CBR
内容分发网络 CDN
存储容灾服务 SDRS
弹性文件服务 SFS
云服务器备份 CSBS
云硬盘备份 VBS
数据快递服务 DES
专属企业存储服务
智能边缘
智能边缘云 IEC
EI 企业智能
EI安视服务
AI开发平台ModelArts
数据湖治理中心 DGC
数据仓库服务 GaussDB(DWS)
企业级AI应用开发专业套件 ModelArts Pro
数据湖探索 DLI
华为HiLens
云搜索服务 CSS
数据接入服务 DIS
表格存储服务 CloudTable
数据湖工厂 DLF
图引擎服务 GES
推荐系统 RES
文字识别 OCR
内容审核 Moderation
图像识别 Image
图像搜索 ImageSearch
人脸识别服务 FRS
对话机器人服务 CBS
视频分析服务 VAS
数据可视化 DLV
视频接入服务 VIS
自然语言处理 NLP
语音交互服务 SIS
知识图谱 KG
医疗智能体 EIHealth
可信智能计算服务 TICS
园区智能体 CampusGo
实时流计算服务 CS
人证核身服务 IVS
IoT物联网
设备接入 IoTDA
IoT物联网
全球SIM联接 GSL
设备发放 IoTDP
IoT开发者服务
IoT边缘 IoTEdge
IoT数据分析
路网数字化服务 DRIS
开发与运维
项目管理 ProjectMan
代码托管 CodeHub
流水线 CloudPipeline
代码检查 CodeCheck
编译构建 CloudBuild
部署 CloudDeploy
云测 CloudTest
发布 CloudRelease
移动应用测试 MobileAPPTest
CloudIDE
Classroom
软件开发平台 DevCloud
开源镜像站 Mirrors
视频
媒体处理 MPC
视频点播 VOD
视频直播 Live
实时音视频 SparkRTC
管理与部署
统一身份认证服务 IAM
云监控服务 CES
应用运维管理 AOM
应用性能管理 APM
云审计服务 CTS
云日志服务 LTS
标签管理服务 TMS
资源管理服务 RMS
应用身份管理服务 OneAccess
专属云
专属计算集群 DCC
专属分布式存储服务 DSS
域名与网站
域名注册服务 Domains
云速建站 CloudSite
企业协同
华为云WeLink
会议
ISDP
解决方案
全栈专属服务
高性能计算 HPC
SAP
游戏云
混合云灾备
快视频
华为工业云平台 IMC
价格
价格原则
成本优化最佳实践
昇腾
昇腾MindX SDK (20.3)
其他
管理控制台
消息中心
产品价格详情
系统权限
我的凭证
客户关联华为云合作伙伴须知
公共问题
宽限期保留期
奖励推广计划
活动
容器
云容器引擎 CCE
云容器实例 CCI
容器镜像服务 SWR
应用编排服务 AOS
容器交付流水线 ContainerOps
应用服务网格 ASM
多云容器平台 MCP
基因容器 GCS
容器洞察引擎 CIE
容器批量计算 BCE
云原生服务中心 OSC
网络
虚拟私有云 VPC
弹性公网IP EIP
弹性负载均衡 ELB
NAT网关 NAT
云专线 DC
虚拟专用网络 VPN
云连接 CC
VPC终端节点 VPCEP
数据库
云数据库 RDS
文档数据库服务 DDS
分布式数据库中间件 DDM
数据复制服务 DRS
数据管理服务 DAS
云数据库 GaussDB(for MySQL)
云数据库 GaussDB NoSQL
云数据库 GaussDB (for openGauss)
数据库和应用迁移 UGO
大数据
MapReduce服务 MRS
应用中间件
应用管理与运维平台 ServiceStage
分布式缓存服务 DCS
分布式消息服务Kafka版
分布式消息服务RabbitMQ版
消息通知服务 SMN
微服务引擎 CSE
云性能测试服务 CPTS
区块链服务 BCS
API网关 APIG
应用魔方 AppCube
分布式消息服务RocketMQ版
多云高可用服务 MAS
可信跨链数据链接服务 TCDAS
企业应用
云桌面 Workspace
云解析服务 DNS
应用与数据集成平台 ROMA Connect
ROMA资产中心 ROMAExchange
ROMA API
鸿源云道
华为乾坤
安全与合规
Web应用防火墙 WAF
漏洞扫描服务 VSS
企业主机安全 HSS
容器安全服务 CGS
数据加密服务 DEW
数据库安全服务 DBSS
态势感知 SA
云堡垒机 CBH
SSL证书管理 SCM
云证书管理服务 CCM
管理检测与响应 MDR
数据安全中心 DSC
威胁检测服务 MTD
DDoS防护 ADS
云防火墙 CFW
应用信任中心 ATC
安全技术与应用
迁移
主机迁移服务 SMS
对象存储迁移服务 OMS
云数据迁移 CDM
智能协作
IdeaHub
企业网络
云管理网络
SD-WAN 云服务
边缘数据中心管理 EDCM
废弃-华为乾坤安全云服务
云通信
语音通话 VoiceCall
消息&短信 MSGSMS
隐私保护通话 PrivateNumber
开发者工具
SDK开发指南
API签名指南
DevStar
HCloud CLI
Terraform
Ansible
云生态
云市场
鲲鹏
昇腾
合作伙伴中心
华为云培训中心
用户服务
帐号中心
费用中心
成本中心
资源中心
企业管理
工单管理
客户运营能力
国际站常见问题
网站备案
支持计划
专业服务
合作伙伴支持计划
更新时间:2021/08/06 GMT+08:00
分享

压缩和转换模型操作

针对您在ModelArts或者本地构建的模型,为获得更高的算力,希望将模型应用于Ascend芯片、ARM或GPU上,此时,您需要将已有模型压缩/转换成相应的格式后,再应用至不同的芯片类型。

ModelArts提供了模型转换功能,即将已有的模型转换成所需格式,以便应用于算力和性能更高的芯片上。

模型转换主要应用场景如下所示:

  • 使用Caffe(.caffemodel格式)或者Tensorflow框架(“frozen_graph”“saved_model”格式)训练的模型,使用转换功能可转换成om格式,转换后的模型可在昇腾(Ascend)芯片上部署运行。
  • 使用Tensorflow框架训练模型(frozen_graph或“saved_model”格式), 使用转换功能可以将模型转换量化成tflite格式,转换后的模型可以在ARM上部署运行。
  • 使用Tensorflow框架训练模型(frozen_graph或“saved_model”格式), 使用转换功能可以将模型转换量化成tensorRT格式, 转换后的模型可以在nvidia p4 GPU上部署运行。

约束限制

  • 模型转换当前只支持三种芯片类型,分别为:Ascend、ARM、GPU。
  • 模型转换当前仅支持原始框架类型为Caffe和Tensorflow的模型转换。当原始框架类型为Caffe时,输入数据类型为FLOAT;当原始框架类型为Tensorflow时,输入数据类型为INT32、BOOL、UINT8、FLOAT。
  • ModelArts提供了转换模板供用户选择,只能选择对应模板进行转换,支持的模板描述,请参见转换模板
  • 现阶段由于tflite和tensorRT支持的算子和量化算子有限,可能存在部分模型转换失败的情况。如果出现转换失败,可以将页面的日志框滑到最后或者去转换输出目录下查看错误日志。
  • 针对用于Ascend芯片的模型转换,其转换限制说明可参见“昇腾开发者社区”约束及参数说明
  • 压缩/转换任务指定的OBS路径,需确保OBS目录与ModelArts在同一区域。
  • 转换后的模型,再导入ModelArts时,需使用模型模板导入
  • 针对“训练管理 > 训练作业 > 预置算法”,只有“yolov3_resnet18”支持模型转换,其他预置算法不支持模型转换功能。针对AI Gallery > 算法”,官方发布的算法,都支持使用模型转换功能。
  • 支持模型文件类型为onnx的模型转换,转换时会先将其转换为Tensorflow框架的FrozenGraphDef格式,然后再转换为om格式。转换工具要求onnx版本为1.6.0,opset为9+。
  • 当原始框架类型为Caffe时,模型文件(.prototxt)和权重文件(.caffemodel)的op name、op type必须保持名称一致(包括大小写)。
  • 当原始框架类型为Caffe时,除了top与bottom相同的layer以外(例如BatchNorm,Scale,ReLU等),其他layer的top名称需要与其name名称保持一致。
  • 当原始框架类型为TensorFlow时,支持FrozenGraphDef格式和SavedModel 格式。如果是SavedModel格式,转换时会先将其转换为FrozenGraphDef格式,然后再转换为om格式。
  • 不支持动态shape的输入,例如:NHWC输入为[?,?,?,3]多个维度可任意指定数值。模型转换时需指定固定数值。
  • 输入数据最大支持四维,转维算子(reshape、expanddim等)不能输出五维。
  • 模型中的所有层算子除const算子外,输入和输出需要满足“dim!=0”
  • 模型转换不支持含有训练算子的模型。
  • 量化(uint8)后的模型不支持模型转换。
  • 模型中的算子只支持2D卷积,暂不支持3D卷积。暂不支持多批量转换batch_normalization_1算子和FusedBatchNorm算子。
  • 只支持Caffe算子清单Tensorflow算子清单中的算子,并需满足算子限制条件。

创建模型压缩/转换任务

  1. 登录ModelArts管理控制台,在左侧导航栏中选择“模型管理 > 压缩/转换”,进入模型转换列表页面。
  2. 单击左上角的“创建任务”,进入任务创建任务页面。
  3. “创建任务”页面,参考,填写转换任务的详细参数。
    1. 填写转换任务的“名称”“描述”信息。
    2. 填写转换任务的具体参数,详细描述请参见表1
      表1 参数说明

      参数

      说明

      输入框架

      目前支持转换的框架有“Caffe”“TensorFlow”

      转换输入目录

      用于转换的模型所在目录,此目录必须为OBS目录,且模型文件的目录需符合ModelArts规范,详情请参见模型输入目录规范

      输出框架

      “输入框架”选择“Caffe”时,“输出框架”支持“MindSpore”

      “输入框架”选择“TensorFlow”时,“输出框架”支持“TFLite”“MindSpore”“TensorRT”

      转换输出目录

      模型转换完成后,根据此参数设置的目录存储模型。输出目录需符合ModelArts规范要求,详情请参见模型输出目录说明

      转换模板

      ModelArts提供了一系列的模板,定义转换功能以及转换过程中所需的参数。当前支持的转换模板详细描述请参见转换模板。转换模板右侧下拉框,将根据您选择的“输入框架”“输出框架”,展现匹配的模板。

      在下拉框选择模板后,下方将呈现此模板对应的高级参数。例如量化精度,方便您可以对模型转换任务进行更高阶的设置。

      不同的转换模板,其对应的高级选项支持的参数不同,每个模板支持的详细参数,请参见转换模板

      图1 创建模型压缩/转换任务
  4. 任务信息填写完成后,单击右下角“立即创建”

    创建完成后,系统自动跳转至“模型压缩/转换列表”中。刚创建的转换任务将呈现在界面中,其“任务状态”“初始化”。任务执行过程预计需要几分钟到十几分钟不等,请耐心等待,当“任务状态”变为“成功”时,表示任务运行完成并且模型转换成功。

    如果“任务状态”变为“失败”,建议单击任务名称进入详情页面,查看日志信息,根据日志信息调整任务的相关参数并创建新的转换任务。

    模型转换成功后,有如下几个使用场景:

    • 可以在HiLens管理控制台,导入转换后的模型,导入后可将模型安装部署至HiLens Kits设备上。
    • 登录“转换输出目录”对应的OBS路径,下载目录中的模型(.om格式),部署至您的设备中。

删除压缩/转换任务

针对运行结束的任务,如果不需要再使用,您可以删除转换任务。其中,“运行中”“初始化”状态中的任务不支持删除操作。

任务删除后,将无法恢复,请谨慎操作。

  • 删除单个

    “模型压缩/转换列表”中,针对需要删除的单个任务,您可以在此任务所在行,单击操作列的“删除”,完成删除操作。

  • 批量删除

    “模型压缩/转换列表”中,勾选多个待删除的任务,然后单击左上角“删除”,完成批量任务的删除操作。

分享:

    相关文档

    相关产品