• 沒有找到結果。

导入/转换 ModelArts 开发模型

9.5 管理模型

9.5.3 导入/转换 ModelArts 开发模型

技能可以抽象地理解为算法模型+逻辑代码。算法模型负责关键的AI推理,逻辑代码负 责处理模型推理的结果。因此在HiLens Studio开发技能时,需要将模型导入HiLens Studio。

本章节介绍如何将在ModelArts开发的模型导入HiLens Studio,以及针对非“om”格 式的模型,如何在HiLens Studio进行模型转换。

前提条件

已在ModelArts在线训练算法模型,训练模型可参见《ModelArts文档》。如果要使用 ModelArts的预置算法,当前华为HiLens平台仅支持转换如下预置算法:

yolov3_resnet18(检测物体类别和位置)

ResNet_v1_50(图像分类)

说明

● 暂不支持导入ModelArts中“自动学习”训练的模型。

● 华为HiLens 只能导入ModelArts中训练的模型文件,不能导入ModelArts的模型。

导入(转换)模型

1. 在HiLens Studio界面左侧,单击 。

页面左侧将显示您在ModelArts训练好的模型列表。

9-16 模型列表

2. 选择待导入、转换的模型,单击“Operation”列的“Apply”。

– 如果是“om”格式的模型,待模型导入成功后,HiLens Studio右下角会提示

“Model imported successfully.”,文件夹“model”下会显示新导入的模 型文件。

– 如果是非“om”格式的模型,会弹出“Convert Model”对话框。

按表9-10填写模型转换的信息,单击“OK”。

9-10 Convert Model 参数说明

参数 说明

Model Path 待转换的模型文件在技能项目文件中的位置。一般将模型导 入至文件夹“save_model”。

Configurati

on 待转换模型的配置文件在技能项目文件中的位置。例如

“save_model/aipp_rgb.cfg”。

Output

Path 模型转换后输出位置。

参数 说明

Type 模型转换的类型,包括“TF-FrozenGraph-To-Ascend-HiLens”、“Caffe to Ascend”。

● “TF-FrozenGraph-To-Ascend-HiLens”

支持将Tensorflow frozen graph模型转换成可在ascend 芯片上运行的模型。

● “Caffe to Ascend”

支持将Caffe模型转换成可在ascend芯片上运行的模型。

Advanced

Options 当模型转换类型为“TF-FrozenGraph-To-Ascend-HiLens”

时,可填写高级选项,包括张量形状、转换输出节点等参数 选项,详情请见表9-11。

9-11 Advanced Options

参数名称 参数说明

Input Tensor

Shape 输入张量形状。若在上文“模型来源”中,选择需要 转换格式的模型(非om格式模型),并且转换类型 是“Tensorflow frozen graph 转 Ascend”或

“Tensorflow SavedModel 转 Ascend”时,需要填 写输入张量形状。

out_nodes 转换输出节点,即指定输出节点,例如

“node_name1:0;node_name1:1;node_name2:0”

,其中“node_name”必须是模型转换前的网络模 型中的节点名称,冒号后的数字表示第几个输出,例如

“node_name1:0”,表示节点名称为

“node_name1”的第0个输出。

input_format 输入数据格式,默认是“NHWC”,如果实际是

“NCHW”的话,需要通过此参数指定“NCHW”。

net_format 优选数据格式,即指定网络算子优先选用的数据格 式,“ND(N=4)”和“5D”。仅在网络中算子的输

参数名称 参数说明 fp16_high_precsio

n 生成高精度模型,指定是否生成高精度“FP16 Davinci”模型。

● 0为默认值,表示生成普通“FP16 Davinci”模 型,推理性能更好。

● 1表示生成高精度“FP16 Davinci”模型,推理精 度更好。

output_type 网络输出数据类型,“FP32”为默认值,推荐分类网 络、检测网络使用;图像超分辨率网络,推荐使用

“UINT8”,推理性能更好。