AI开发平台ModelArtsAI开发平台ModelArts

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

ResNet_v2_101(图像分类)

概述

该模型为ResNet系列,深度为101层,此模型基于Deep Residual Learning for Image Recognition中提出的模型结构实现。可以用图像分类任务,比如猫狗分类、花卉分类等等。用户提供一系列带有标注的数据集,该算法会载入在ImageNet-1000上的预训练模型,在用户数据集上做迁移学习。训练后生成的模型可直接在ModelArts平台部署为在线服务或批量服务,同时支持使用CPU、GPU或Ascend 310进行推理。

ResNet_v2_101和ResNet_v2_50的区别,在于模型深度不同。

单击此处订阅算法。

训练

  • 算法基本信息
    • 适用场景:图像分类
    • 支持的框架引擎:Tensorflow-1.13.1-python3.6-horovod
    • 算法输入:
      • ModelArts数据管理平台发布的图像分类数据集(数据集必须设置“训练验证比例”),建议用户以8:2或9:1的比例进行切分,即“训练验证比例”设置为0.8或0.9。
      • 在ImageNet上的预训练模型。
    • 算法输出:
      • 用于TF-Serving推理的saved_model模型。
      • 用于转换Ascend格式模型的Frozen_PB模型。
  • 训练参数说明

    名称

    默认值

    类型

    是否必填

    是否可修改

    描述

    task_type

    image_classification_v2

    string

    适用场景。

    model_name

    resnet_v2_101

    string

    模型名称。

    do_train

    TRUE

    bool

    是否做训练,默认训练。

    do_eval_along_train

    TRUE

    bool

    是否边训练边验证,默认值为True,表示边训练边验证。

    variable_update

    horovod

    string

    参数更新方式,默认horovod。

    learning_rate_strategy

    0.002

    string

    训练的学习率策略。"10:0.001,20:0.0001"代表0-10个epoch学习率0.001,10-20epoch学习率0.0001。如果不指定epoch,会根据验证精度情况自动调整学习率,并当精度没有明显提升时,训练停止。

    batch_size

    32

    int

    每步训练的图片数量(单卡)。

    eval_batch_size

    32

    int

    每步验证的图片数量(单卡)。

    evaluate_every_n_epochs

    1

    float

    每训练n个epoch做一次验证。

    save_model_secs

    60

    int

    保存模型的频率(单位:s)。

    save_summary_steps

    10

    int

    保存summary的频率(单位:步)。

    log_every_n_steps

    10

    int

    打印日志的频率(单位:步)。

    do_data_cleaning

    TRUE

    bool

    是否做数据清洗,数据格式异常会导致训练失败,建议开启,保证训练稳定性。数据量过大时,数据清洗可能耗时较久,可自行线下清洗(支持BMP、JPEG、PNG格式,RGB三通道)。建议用JPEG格式数据。默认开启,设置为False表示关闭。

    use_fp16

    TRUE

    bool

    是否使用混合精度,混合精度可以加速训练,但是可能会造成一点精度损失,若对精度无极严格的要求,建议开启。默认启用,设置为False表示关闭。

    xla_compile

    TRUE

    bool

    是否开启xla编译,加速训练,默认启用,设置为False表示关闭。

    data_format

    NCHW

    string

    输入数据类型,NHWC表示channel在最后,NCHW表示channel在最前,默认值NCHW(速度有提升)。

    best_model

    TRUE

    bool

    是否在训练过程中保存并使用精度最高的模型,而不是最新的模型。默认值True,保存最优模型。在一定误差范围内,最优模型会保存最新的高精度模型。设置为False表示关闭。

  • 训练输出文件

    训练完成后的输出文件如下。

    |- om    
      |- model      
        |- index      
          |- customize_service_d310.py  
    |- model    
      |- variables      
        |- variables.data-00000-of-00001      
        |- variables.index    
      |- customize_service.py    
      |- index    
      |- config.json    
      |- saved_model.pb  
    |- frozen_graph    
      |- insert_op_conf.cfg    
      |- model.pb  
    |- checkpoint  
    |- model.ckpt-xxx  
    |- ...  
    |- best_checkpoint  
    |- best_model.ckpt-xxx  
    |- ...  
    |- events...  
    |- graph.pbtxt

Ascend 310推理

  • 模型转换
    • 转换模板:TF-FrozenGraph-To-Ascend-C32
    • 转换输入目录:选择“训练输出目录”中的frozen_graph
    • 转换输出目录:选择“训练输出目录”中的om/model
    • 输入张量形状:images:1,299,299,3
    • 输入数据格式:NHWC

      其他参数使用默认值。

  • 模型导入
    • 从模板中选择:ARM-Ascend模板
    • 模型目录:选择“训练输出目录”中的om/model
    • 输入输出模式:预置图像处理模式

案例指导

GPU训练+Ascend 310推理,可参考图像分类案例

分享:

    相关文档

    相关产品

关闭导读