目标检测后处理
基类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异常。
父主题： 模型后处理