Views
No views yet
| Model | Top1 Acc(%) | GPU Inference Time (ms) [Regular Mode / High-Performance Mode] | CPU Inference Time (ms) [Regular Mode / High-Performance Mode] | Model Storage Size (M) |
|---|---|---|---|---|
| RT-DETR-L_wired_table_cell_det | 82.7 | 35.00 / 10.45 | 495.51 / 495.51 | 124M |
1import requests
2from PIL import Image
3from transformers import AutoImageProcessor, AutoModelForObjectDetection
4
5model_path = "PaddlePaddle/RT-DETR-L_wired_table_cell_det_safetensors"
6model = AutoModelForObjectDetection.from_pretrained(model_path)
7image_processor = AutoImageProcessor.from_pretrained(model_path)
8
9image = Image.open(requests.get("https://paddle-model-ecology.bj.bcebos.com/paddlex/imgs/demo_image/table_recognition.jpg", stream=True).raw)
10inputs = image_processor(images=image, return_tensors="pt")
11
12outputs = model(**inputs)
13results = image_processor.post_process_object_detection(outputs, target_sizes=[image.size[::-1]])
14for result in results:
15 for score, label_id, box in zip(result["scores"], result["labels"], result["boxes"]):
16 score, label = score.item(), label_id.item()
17 box = [round(i, 2) for i in box.tolist()]
18 print(f"{model.config.id2label[label]}: {score:.2f} {box}")