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.744 | 0.777 | 0.905 | 0.910 | 0.823 | 0.581 | 0.727 | 0.721 | 0.575 | 0.647 | 0.827 | 0.525 | 0.770 |
1import requests
2from PIL import Image
3from transformers import AutoImageProcessor, AutoModelForObjectDetection
4
5model_path="PaddlePaddle/PP-OCRv5_mobile_det_safetensors"
6model = AutoModelForObjectDetection.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_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_object_detection(outputs, target_sizes=inputs["target_sizes"])
14
15for result in results:
16 print(result["boxes"])
17 print(result["scores"])