使用预置算法训练所得的模型,需通过转换功能,转成适用于Ascend芯片的格式,即
“om”格式。
步骤1 在ModelArts管理控制台,选择“模型管理>压缩/转换”。
步骤2 在模型转换作业列表页面,单击“创建任务”。
步骤3 在创建转换任务页面按表7-5设置参数。
最佳实践 7 口罩识别技能开发(ModelArts+华为 HiLens)
图7-12 模型转换
表7-5 模型转换参数说明
参数 说明 推荐填写
名称 模型转换任务的名称。 本样例可设置名称“convert-mask-detection”。
描述 模型转换任务的描述。
-输入框架 当“输入框架”选择“Caffe”时,
“输出框架”支持
“MindSpore”。
当“输入框架”选择
“TensorFlow”时,“输出框架”
支持“TFLite”、
“MindSpore”、“TensorRT”。
本样例选择“TensorFlow”。
转换输入目 录
用于转换的模型所在目录,此目录
必须为OBS目录。 本样例选择训练作业输出路径 下的“frozen_gragh”目录,
即“mask/model-output/
frozen_gragh”。
输出框架 根据实际情况选择模型转换后的框 架。
本样例选择“MindSpore”。
转换输出目
录 模型转换完成后,根据此参数设置
的目录存储模型。 本样例选择训练作业输出路
径,即“mask/model-output”。
最佳实践 7 口罩识别技能开发(ModelArts+华为 HiLens)
参数 说明 推荐填写 转换模板 ModelArts提供了一系列的模板,
定义转换功能以及转换过程中所需
本样例选择“TF-FrozenGraph-To-Ascend-HiLens”,即将 TensorFlow的frozen graph格 式的模型转换成可在昇腾芯片 上推理的格式。
高级选项 “输入张量形状”、“输入数据格 式”等参数。
“输入张量形状”需根据训练 作业中的“input_shape”参数 进行设置,本示例设置为
“images:1,352,640,3”。
“images”代表模型输入节 点,“1”表示batch size,
“3”表示channels数量,其 中,“images”、“1”和
“3”都是固定的,不支持修 改,“352”和“640”根据训 练作业中的“input_shape”参 数进行设置。“images:
1,352,640,3”中间用英文逗号 隔开,不能包含空格。其他
步骤1 登录华为HiLens管理控制台,在左侧导航栏中选择“技能开发>模型管理”,进入
“模型管理”页面。
步骤2 在“模型管理”页面,单击右上角的“导入(转换)模型”。
步骤3 在“导入模型”页面,然后填写参数,信息确认无误后单击“确定”完成导入。
● “名称”:输入“mask-recognition”。
● “版本”:输入“1.0.0”。
● “描述”:输入导入模型的简短描述。
最佳实践 7 口罩识别技能开发(ModelArts+华为 HiLens)
● “模型来源”:单击“从ModelArts导入”,在右侧下拉框中选择“OM(从转换 任务中获取)”,然后在下方转换任务列表中勾选之前在步骤5:模型转换中转换 的模型“convert-mask-detection”。
图7-14 导入模型
步骤2 在华为HiLens管理控制台的左侧导航栏中选择“技能开发>技能管理”,进入技能列 表。
步骤3 在“技能管理”页面,单击右上角“新建技能”,进入“创建技能”页面。
步骤4 在“创建技能”页面,在“技能模板”中选择“使用空模板”后,填写基本信息和技 能内容。
基本信息 如图7-15所示。
● “技能模板”:选择“使用空模板”。
● “技能名称”:输入“Mask_Detection”。
● “技能版本”:输入“1.0.0”。
● “适用芯片”:默认为“Ascend310”。
● “检验值”:根据“index.py”文件中init函数参数的值的代码,这里检验值应输 入“mask”。此处的检验值和init函数参数值应保持一致。
● “应用场景”:选择“其他”,文本框中输入“口罩识别”。
● “技能图标”:上传技能图片。
● “OS平台”:选择“Linux”系统。
● “摘要”:输入技能摘要。
● “描述”:输入技能描述。
最佳实践 7 口罩识别技能开发(ModelArts+华为 HiLens)
图7-15 填写基本信息
技能内容
根据您的模型和逻辑代码情况,填写技能内容,详细参数说明请参见图7-16。
● “技能格式”:选择“技能包”。
● “运行时语言”:选择“Python3.7”。
● “代码执行文件”:输入“index.py”。
● “模型”:单击加号,您可以在弹出框中,选择导入的模型“mask-detection”。
● “代码上传方式”:选择“在线编辑”,在代码编辑框中直接编辑代码,可单击 技能代码下载技能代码,复制所下载的文件夹中的“index.py”和 “utils.py”文 件内容。 “utils.py”文件可通过代码编辑框左上角“文件>创建同级文件”来创 建。
最佳实践 7 口罩识别技能开发(ModelArts+华为 HiLens)
图7-16 技能内容
步骤5 基本信息和技能内容填写完成后,您可以在界面右侧查看其配置参数值,如果某个字 段填写错误,在右侧会显示一个小红叉。
确认信息无误后,单击“确定”完成技能创建。
----结束