TahaDouaji
/
detr-doc-table-detection
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Improve code snippet

This commit is contained in:
Niels Rogge 2022-08-05 13:31:07 +00:00 committed by huggingface-web
parent a3e4b9a10c
commit d60c294e85
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -24,6 +24,7 @@ model = DetrForObjectDetection.from_pretrained('TahaDouaji/detr-doc-table-detect
inputs = feature_extractor(images=image, return_tensors="pt") inputs = feature_extractor(images=image, return_tensors="pt")
outputs = model(**inputs) outputs = model(**inputs)
logits = outputs.logits # convert outputs (bounding boxes and class logits) to COCO API
bboxes = outputs.pred_boxes target_sizes = torch.tensor([image.size[::-1]])
results = feature_extractor.post_process(outputs, target_sizes=target_sizes)[0]
``` ```