分类后处理
基类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异常。
父主题: 模型后处理