关键点检测后处理
基类KeypointPostProcessBase。
支持的模型后处理
模型后处理 |
说明 |
|---|---|
OpenPosePostProcess |
OpenPost后处理。 |
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异常。
父主题： 模型后处理