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