分类后处理

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异常。

父主题： 模型后处理

目标检测后处理

基类ObjectPostProcessBase。

支持的模型后处理

模型后处理	说明
Yolov3PostProcess	YOLOv3后处理。
SsdMobilenetv1FpnPostProcess	SsdMobilenetv1Fpn后处理。
SsdMobilenetFpn_MindsporePost	SsdMobilenetFpn MindSpore后处理。
Ssdvgg16PostProcess	Ssdvgg16后处理。
FasterRcnnPostProcess	FasterRcnn后处理。
MaskRcnnMindsporePost	MaskRcnnMindspore后处理。

统一的构造函数

postProcessName(config_data, label_path)

postProcessName(config_path, label_path)

例如：Yolov3PostProcess(config_data, label_path)，Yolov3PostProcess(config_path, label_path)。

构造函数输入参数说明

参数名	类型	说明
config_data	dict	配置数据。
label_path	str	标签路径。
config_path	str	配置路径。

构造函数输出参数说明

对应后处理对象。

统一的后处理函数

process(tensor, params)

process(tensor, param)

后处理函数输入参数说明

参数名	类型	说明
tensor	List[base.Tensor]	Tensor列表。
params	List[base.ResizedImageInfo]	ResizedImageInfo列表。
param	base.ResizedImageInfo	ResizedImageInfo对象。

后处理函数输出参数说明

list[list[ObjectInfo]]

抛异常接口

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

父主题： 模型后处理

分割后处理

基类SemanticSegPostProcessBase。

支持的模型后处理

模型后处理	说明
Deeplabv3PostProcess	Deeplabv3后处理。
Deeplabv3PytorchPost	Deeplabv3 pytorch后处理。
Deeplabv3TensorflowPost	Deeplabv3 TensorFlow后处理。
Deeplabv3MindsporePost	Deeplabv3 MindSpore后处理。
UNetMindSporePostProcess	UNet MindSpore后处理。

统一的构造函数

postProcessName(config_data, label_path)

postProcessName(config_path, label_path)

例如：Deeplabv3PostProcess (config_data, label_path)，Deeplabv3PostProcess(config_path, label_path)。

构造函数输入参数说明

参数名	类型	说明
config_data	dict	配置数据。
label_path	str	标签路径。
config_path	str	配置路径。

构造函数输出参数说明

对应的后处理对象。

统一的后处理函数

process(tensor, params)

process(tensor, param)

后处理函数输入参数说明

参数名	类型	说明
tensor	List[base.Tensor]	Tensor列表。
params	List[base.ResizedImageInfo]	ResizedImageInfo列表。
param	base.ResizedImageInfo	ResizedImageInfo对象。

后处理函数输出参数说明

list[SemanticSegInfo]

抛异常接口

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

父主题： 模型后处理

文本生成后处理

基类TextGenerationPostProcessBase。

支持的模型后处理

模型后处理	说明
CrnnPostProcess	Crnn后处理。
TransformerPostProcess	Transformer后处理。

统一的构造函数

postProcessName(config_data, label_path)

postProcessName(config_path, label_path)

例如：CrnnPostProcess(config_data, label_path)，CrnnPostProcess(config_path, label_path)。

构造函数输入参数说明

参数名	类型	说明
config_data	dict	配置数据。
label_path	str	标签路径。
config_path	str	配置路径。

构造函数输出参数说明

对应的后处理对象。

统一的后处理函数

process(tensor, params)

process(tensor, param)

后处理函数输入参数说明

参数名	类型	说明
tensor	List[base.Tensor]	Tensor列表。
params	List[base.ResizedImageInfo]	ResizedImageInfo列表。
param	base.ResizedImageInfo	ResizedImageInfo对象。

后处理函数输出参数说明

list[TextsInfo]

抛异常接口

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

父主题： 模型后处理

在线提单