ComPDF
联系销售
文档DemoDownloadFAQ更新日志

设置 DocumentAI 模型 ​

概述 ​

在使用 OCR、版面分析、表格识别、PDF 转可搜索 PDF 或 PDF 转 OFD 之前，需要先设置 DocumentAI 模型路径。

CPDF_SetDocumentAIModel 支持通过 gpu_id 参数指定 AI 模型使用的 GPU 设备编号。当 gpu_id 为 -1 时，表示禁用 GPU 加速。

示例 ​

c
CSDKErrorCode code = CPDF_SetDocumentAIModel(CPDF_TEXT("path/documentai.model"), -1);
if (code != e_CErrSuccess) {
    return code;
}

设置 AI 模型实例数量 ​

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

c
CPDF_SetDocumentAIModelCount(1, 1);

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

使用自定义 AI 引擎（SDK v4.1.0+） ​

该选项仅适用于 SDK v4.1.0 及以上版本。如果您希望使用自有模型或第三方服务来完成 OCR、版面分析或表格识别，而不是使用内置的 DocumentAI 模型，SDK 提供了 CConvertCallback 回调接口，允许您以 JSON 形式传入识别结果。详情请参见 4.11 通过回调使用自定义 AI 模型。当所需能力全部由自定义回调覆盖时，可以不调用 CPDF_SetDocumentAIModel。