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