Skip to content

设置 DocumentAI 模型

概述

当需要使用 OCR、版面分析、表格识别、PDF 转可搜索 PDF 或 PDF 转 OFD 等能力时,需要先设置 DocumentAI 模型路径。

SetDocumentAIModel 接口支持传入 gpu_id 参数,用于指定 AI 模型使用的 GPU 设备编号;当 gpu_id-1 时,表示不启用 GPU。

示例

c++
LibraryManager::SetDocumentAIModel("path/documentai_v4.model", -1);

设置 AI 模型实例数量

如果需要控制版面分析和表格识别模型实例数量,可调用以下接口:

c++
LibraryManager::SetDocumentAIModelCount(1, 1);

其中,第一个参数表示版面分析模型实例数量,第二个参数表示表格识别模型实例数量。