Skip to content

设置 DocumentAI 模型

概述

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

setDocumentAIModel 支持可选的 gpuId 参数,用于指定 AI 模型使用的 GPU 设备编号。当 gpuId-1 时,表示不启用 GPU 加速。

设置 AI 模型实例数量

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

示例

js
const modelPath = "/path/to/documentai.model";
sdk.setDocumentAIModelCount(1, 1);
const code = sdk.setDocumentAIModel(modelPath, -1);
if (code !== 0) {
  throw new Error(`setDocumentAIModel failed: ${code}`);
}

转换任务完成后,如果加载过 DocumentAI 模型,可以调用 releaseDocumentAIModel() 释放模型资源。

js
sdk.releaseDocumentAIModel();