TrOCR

Microsoft의 TrOCR을 전용 배포 또는 자체 호스팅 추론에서 텍스트 인식에 사용

TrOCR는 Microsoft의 트랜스포머 기반 OCR 모델입니다. 줄 수준의 텍스트 인식을 위해 학습되었으므로, 최상의 결과를 얻으려면 입력을 하나의 텍스트 영역으로 잘라내세요.

TrOCR는 Serverless Cloud API에서 사용할 수 없습니다. 다음에서 실행하세요: 전용 배포 또는 자가 호스팅 추론.

코드 예시

1

API 키 받기

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

export ROBOFLOW_API_KEY="your-key-here"

2

의존성 설치

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

pip install -U requests opencv-python supervision

3

모델 실행

설정 URL 전용 배포 URL 또는 로컬 Inference 서버로.

import base64
import os
import cv2
import requests
import supervision as sv

URL = "https://your-deployment.roboflow.cloud"
image = sv.load_image_from_url("https://media.roboflow.com/inference/license_plate_1.jpg")
_, 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 서버용.

• 당신의 전용 배포 비공개 엔드포인트용 URL.

Inference(자가 호스팅)에서 사용

TrOCR는 다음에서 제공됩니다 Roboflow Inference 자신의 하드웨어에서 실행. 로컬 서버를 시작한 다음, 다음을 사용하여 모델을 선택하세요: model 공유 OCR 엔드포인트의 인수:

pip install inference-cli
inference 서버 시작  # http://localhost:9001에서 서비스

from inference_sdk import InferenceHTTPClient

client = InferenceHTTPClient(api_url="http://127.0.0.1:9001")

result = client.ocr_image(inference_input="./serial_number.png", model="trocr")
print(result)

TrOCR는 잘라낸 단일 줄의 인쇄 텍스트에서 가장 좋은 성능을 보입니다. 전송하기 전에 각 텍스트 영역을 잘라내세요. 다른 일부 OCR 모델과 달리, TrOCR는 잘라내지 않은 이미지나 여러 줄로 된 이미지를 잘 처리하지 못합니다.