【Azure Document intelligence/python】- カスタムモデルをAPIで実行する方法

執筆日

2024/10/25

前提

• Azure Document intelligenceでカスタムモデルを構築済みであること
• ローカルにドキュメントがあること

code

1. 以下のコマンドを実行し、ライブラリーをinstall

pip install azure-ai-formrecognizer

2. 以下を実行

main.py

from azure.ai.formrecognizer import DocumentAnalysisClient
from azure.core.credentials import AzureKeyCredential

# Azure Document IntelligenceのエンドポイントとAPIキーを設定
endpoint = "<endpoint>"
api_key = "<api key>"

# DocumentAnalysisClientを作成します
client = DocumentAnalysisClient(endpoint, AzureKeyCredential(api_key))

# カスタムモデルのIDを設定します
model_id = "<model id>"

# 解析するローカルファイルのパスを設定します
file_path = "<ファイルまでのパス>"

# ローカルファイルを開いて解析します
with open(file_path, "rb") as f:
    poller = client.begin_analyze_document(model_id, document=f)
    result = poller.result()


# 解析結果を表示します（Tablesを作成していないので、Filedだけ出力しています。）
for document_index, document in enumerate(result.documents):
    for name, field in document.fields.items():
        print(f"Field '{name}': {field.value} (confidence: {field.confidence})")

カスタムモデルのID取得方法

1. 該当のAzure Document intelligenceを開き、Studioを立ち上げる
2. 作成したカスタムモデルの種類をクリックする
   
3. 該当のプロジェクトをクリックする
   
4. ページ遷移後に、左タブの「Models」をクリックする
   
5. Model ID　を取得する