TrOCR

Dedicated Deployment 또는 self-hosted Inference에서 Microsoft의 TrOCR을 사용해 텍스트 인식을 수행합니다.

TrOCR는 Microsoft의 transformer 기반 OCR 모델입니다. 행 수준 텍스트 인식을 위해 학습되었으므로, 최상의 결과를 위해 입력을 단일 텍스트 영역으로 잘라내세요.

TrOCR는 Serverless Hosted API에서 사용할 수 없습니다. 다음에서 실행하세요 Dedicated Deployment 또는 자체 호스팅 Inference.

코드 샘플

1

API Key를 받으세요

Roboflow 계정을 만들고, 다음에서 키를 찾으세요: Roboflow API 설정 페이지 그리고 이를 셸에서 사용할 수 있도록 설정하세요:

export ROBOFLOW_API_KEY="your-key-here"

2

종속성을 설치하세요

다음 패키지는 이미지를 가져와 API를 호출합니다:

pip install requests opencv-python

3

모델을 실행하세요

설정 URL 를 Dedicated Deployment URL 또는 로컬 Inference server에.

import base64
import os
import cv2
import numpy as np
import requests

URL = "https://your-deployment.roboflow.cloud"
content = requests.get("https://media.roboflow.com/inference/license_plate_1.jpg").content
image = cv2.imdecode(np.frombuffer(content, np.uint8), cv2.IMREAD_COLOR)
_, buffer = cv2.imencode(".jpg", image)
image_base64 = base64.b64encode(buffer).decode("utf-8")

response = requests.post(
    f"{URL}/ocr/trocr",
    json={
        "api_key": os.environ["ROBOFLOW_API_KEY"],
        "image": {"type": "base64", "value": image_base64},
    },
)
print(response.json()["result"])

위의 코드는 인식된 텍스트를 터미널에 출력합니다:

총계

추론 속도

다음 기준으로 측정한 지연 시간 Roboflow Inference 1x NVIDIA L4에서, 배치 크기 1로, 워밍업 이후 평균값입니다.

모델	지연 시간(ms)
trocr	114.4

TrOCR는 잘라낸 단일 텍스트 줄을 인식하므로, 이는 전체 페이지가 아니라 한 줄 크롭에 대한 지연 시간입니다.

설정 URL 을 배포 대상에 맞게 설정하세요:

• http://localhost:9001 로컬 Inference 서버용입니다.

• 귀하의 Dedicated Deployment 비공개 엔드포인트용 URL입니다.