下载
中文
注册

文本生成后处理

基类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异常。