能否通过Python API实现Document AI处理器数据集创建及文档导入?
关于Document AI Python API创建数据集及导入文档的说明
- 目前Google Cloud Document AI的Python官方API暂不直接支持创建数据集和导入文档到数据集的操作,这两个功能当前主要通过Cloud Console(控制台)或gcloud命令行工具实现。
- 你已掌握的创建处理器、训练处理器的Python API操作是官方支持的,但数据集管理相关的Python SDK接口尚未公开。
- 若需通过代码自动化完成这些操作,可考虑调用gcloud命令行工具的对应指令,在Python中通过
subprocess模块执行,示例代码如下:import subprocess # 创建数据集(替换占位参数为实际值) subprocess.run([ "gcloud", "document-ai", "datasets", "create", "你的数据集名称", "--location=你的区域", "--processor-type=处理器类型", "--display-name=数据集显示名称" ], check=True) # 导入文档到数据集 subprocess.run([ "gcloud", "document-ai", "datasets", "import-documents", "你的数据集名称", "--location=你的区域", "--input-pattern=gs://你的存储桶路径/*.pdf", "--mime-type=application/pdf" ], check=True)
注:使用此方式需确保本地环境已配置好gcloud CLI并完成授权,且账号具备对应的Document AI操作权限。
内容的提问来源于stack exchange,提问作者Augusto Firmo
相关产品推荐
相关产品推荐

