Views
No views yet
| Handwritten Chinese | Handwritten English | Printed Chinese | Printed English | Traditional Chinese | Ancient Text | Japanese | General Scenario | Pinyin | Rotation | Distortion | Artistic Text | Average |
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| 0.4166 | 0.4944 | 0.8605 | 0.8753 | 0.7199 | 0.5786 | 0.7577 | 0.5570 | 0.7703 | 0.7248 | 0.8089 | 0.5398 | 0.8015 |
1import requests
2from PIL import Image
3from transformers import AutoImageProcessor, AutoModelForTextRecognition
4
5model_path = "PaddlePaddle/PP-OCRv5_mobile_rec_safetensors"
6model = AutoModelForTextRecognition.from_pretrained(model_path, device_map="auto")
7image_processor = AutoImageProcessor.from_pretrained(model_path)
8
9image = Image.open(requests.get("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/general_ocr_rec_001.png", stream=True).raw).convert("RGB")
10inputs = image_processor(images=image, return_tensors="pt").to(model.device)
11outputs = model(**inputs)
12
13results = image_processor.post_process_text_recognition(outputs)
14
15for result in results:
16 print(result)