Skip to main content

model(modelPath, deviceId)

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)

函数功能

加载模型

输入参数说明

参数名类型说明
modelPathstr模型路径。
deviceIdInt加载的deviceId。

输出参数说明

Model对象。

父主题: 模型推理

分类后处理

基类ClassPostProcessBase。

支持的模型后处理

模型后处理说明
Resnet50PostProcessResNet-50后处理。

构造函数

Resnet50PostProcess(config_data, label_path)
Resnet50PostProcess(config_path, label_path)

Resnet50PostProcess输入参数说明

参数名类型说明
config_datadict配置数据。
label_pathstr标签路径。
config_pathstr配置路径。

Resnet50PostProcess输出参数说明

Resnet50PostProcess对象。

统一的后处理函数

process(tensor)

后处理函数输入参数说明

参数名类型说明
tensorList[base.Tensor]Tensor列表。

后处理函数输出参数说明

list[list[ClassInfo]]

抛异常接口

传入Tensor失败,拷贝至Host失败,抛出runtime异常。

父主题: 模型后处理

目标检测后处理

基类ObjectPostProcessBase。

支持的模型后处理

模型后处理说明
Yolov3PostProcessYOLOv3后处理。
SsdMobilenetv1FpnPostProcessSsdMobilenetv1Fpn后处理。
SsdMobilenetFpn_MindsporePostSsdMobilenetFpn MindSpore后处理。
Ssdvgg16PostProcessSsdvgg16后处理。
FasterRcnnPostProcessFasterRcnn后处理。
MaskRcnnMindsporePostMaskRcnnMindspore后处理。

统一的构造函数

postProcessName(config_data, label_path)
postProcessName(config_path, label_path)

例如:Yolov3PostProcess(config_data, label_path),Yolov3PostProcess(config_path, label_path)。

构造函数输入参数说明

参数名类型说明
config_datadict配置数据。
label_pathstr标签路径。
config_pathstr配置路径。

构造函数输出参数说明

对应后处理对象。

统一的后处理函数

process(tensor, params)
process(tensor, param)

后处理函数输入参数说明

参数名类型说明
tensorList[base.Tensor]Tensor列表。
paramsList[base.ResizedImageInfo]ResizedImageInfo列表。
parambase.ResizedImageInfoResizedImageInfo对象。

后处理函数输出参数说明

list[list[ObjectInfo]]

抛异常接口

传入Tensor失败,拷贝至Host失败,抛出runtime异常。

父主题: 模型后处理

分割后处理

基类SemanticSegPostProcessBase。

支持的模型后处理

模型后处理说明
Deeplabv3PostProcessDeeplabv3后处理。
Deeplabv3PytorchPostDeeplabv3 pytorch后处理。
Deeplabv3TensorflowPostDeeplabv3 TensorFlow后处理。
Deeplabv3MindsporePostDeeplabv3 MindSpore后处理。
UNetMindSporePostProcessUNet MindSpore后处理。

统一的构造函数

postProcessName(config_data, label_path)
postProcessName(config_path, label_path)

例如:Deeplabv3PostProcess (config_data, label_path),Deeplabv3PostProcess(config_path, label_path)。

构造函数输入参数说明

参数名类型说明
config_datadict配置数据。
label_pathstr标签路径。
config_pathstr配置路径。

构造函数输出参数说明

对应的后处理对象。

统一的后处理函数

process(tensor, params)
process(tensor, param)

后处理函数输入参数说明

参数名类型说明
tensorList[base.Tensor]Tensor列表。
paramsList[base.ResizedImageInfo]ResizedImageInfo列表。
parambase.ResizedImageInfoResizedImageInfo对象。

后处理函数输出参数说明

list[SemanticSegInfo]

抛异常接口

传入Tensor失败,拷贝至Host失败,抛出runtime异常。

父主题: 模型后处理

在线提单