设置 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();