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 模型回调

示例

java
int gpuId = 0;
LibraryManager.setDocumentAIModel("path/model", gpuId);