model_postprocessing.md

模型后处理

总体说明

接口使用流程请参考后处理。

支持的型号

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

分类后处理

基类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后处理。
SsdMobilenetFpnMindsporePost	SsdMobilenetFpnMindSpore后处理。
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。

支持的模型后处理

模型后处理	说明
Deeplabv3Post	Deeplabv3后处理。
UNetMindSporePostProcess	UNet MindSpore后处理。

统一的构造函数

postProcessName(config_data, label_path)

postProcessName(config_path, label_path)

例如：Deeplabv3Post (config_data, label_path)，Deeplabv3Post(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)

后处理函数输入参数说明

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

后处理函数输出参数说明

list[TextsInfo]

抛异常接口

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

文本检测后处理

基类TextObjectPostProcessBase。

支持的模型后处理

模型后处理	说明
CtpnPostProcess	Ctpn后处理。
PSENetPostProcess	PSEnet后处理。

统一的构造函数

postProcessName(config_data, label_path)

postProcessName(config_path, label_path)

例如：CtpnPostProcess (config_data, label_path)，CtpnPostProcess (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[TextObjectInfo]]

抛异常接口

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

关键点检测后处理

基类KeypointPostProcessBase。

支持的模型后处理

模型后处理	说明
OpenPosePostProcess	OpenPose后处理。
HigherHRnetPostProcess	HigherHRnet后处理。

构造函数

postProcessName(config_data, label_path)

postProcessName(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[KeyPointDetectionInfo]]

抛异常接口

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