Skip to content

设置 DocumentAI 模型

概述

在使用 OCR、版面分析、表格识别之前,需要先设置 DocumentAI 模型路径。

setDocumentAIModel 支持可选的 gpuId 参数,用于指定 AI 模型的 GPU 设备索引。当 gpuId-1 时,表示禁用 GPU 加速。

设置 AI 模型实例数

如果需要控制版面分析和表格识别的模型实例数量,请调用相应的接口来设置模型实例数。

使用您自己的 AI 引擎(SDK v4.1.0+)

此选项仅在 SDK v4.1.0 或更高版本中可用。您可以跳过 setDocumentAIModel,通过 ConvertCallback 上的回调插入自己的 OCR、版面分析或表格识别引擎。详情请参见 3.11 通过回调使用自定义 AI 模型

示例

objective-c
[LibraryManager setDocumentAIModel:@"path/documentai_v4.model"];
[LibraryManager setDocumentAIModel:@"path/documentai_v4.model" gpuId:-1];
[LibraryManager setDocumentAIModelCount:1 tableModelCount:1];