model_gear()
input_shape(index)
函数功能
获得模型输入的对应Tensor的数据shape信息。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| index | int | 模型输入的索引（第index个输入）。 |
返回参数说明
返回对应输入的Tensor的shape信息，List[int]类型。
父主题： Model类
output_shape(index)
函数功能
获得模型输出的对应Tensor的数据shape信息。
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| index | int | 模型输出的索引（第index个输出）。 |
返回参数说明
返回对应输出的Tensor的shape信息，List[int]类型。
父主题： Model类
model_gear()
函数功能
获得模型支持的动态输入档位信息，支持动态Batch、动态分辨率和分档动态维度模型。
返回参数说明
返回模型支持的输入Tensor档位信息，List[List[int]]类型。
返回数据结构为空，表示获取分档信息失败或者模型不属于函数功能中描述的三种模型之一，具体的原因可查看报错信息确认。
例如，模型为动态分辨率模型，List[int]为某一档的高、宽值。
父主题： Model类
Model类的属性列表
| 属性名 | 说明 | 备注 |
|---|---|---|
| input_format | 获得模型输入的数据组织形式。 | VisionDataFormat类型，NHWC或者NCHW。 |
| input_num | 获得模型的输入个数。 | int类型。 |
| output_num | 获得模型的输出个数。 | int类型。 |
父主题： Model类
model(modelPath, deviceId)
函数功能
加载模型
输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| modelPath | str | 模型路径。 |
| deviceId | Int | 加载的deviceId。 |
输出参数说明
Model对象。
父主题： 模型推理
分类后处理
基类ClassPostProcessBase。
支持的模型后处理
| 模型后处理 | 说明 |
|---|---|
| Resnet50PostProcess | ResNet-50后处理。 |
构造函数
Resnet50PostProcess(config_data, label_path)
Resnet50PostProcess(config_path, label_path)
Resnet50PostProcess输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| config_data | dict | 配置数据。 |
| label_path | str | 标签路径。 |
| config_path | str | 配置路径。 |
Resnet50PostProcess输出参数说明
Resnet50PostProcess对象。
统一的后处理函数
process(tensor)
后处理函数输入参数说明
| 参数名 | 类型 | 说明 |
|---|---|---|
| tensor | List[base.Tensor] | Tensor列表。 |
后处理函数输出参数说明
list[list[ClassInfo]]
抛异常接口
传入Tensor失败，拷贝至Host失败，抛出runtime异常。
父主题： 模型后处理
在线提单