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

Update README.md

This commit is contained in:
Taha Douaji 2023-01-03 06:44:07 +00:00 committed by huggingface-web
parent e05068a73a
commit 7207383528
1 changed files with 2 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
README.md
View File

@ -1,17 +1,3 @@
---
tags:
- object-detection
---
## Model description
detr-doc-table-detection is a model trained to detect both **Bordered** and **Borderless** tables in documents, based on [facebook/detr-resnet-50](https://huggingface.co/facebook/detr-resnet-50)
## Training data
The model was trained on ICDAR2019 Table Dataset
### How to use
```python ```python
from transformers import DetrImageProcessor, DetrForObjectDetection from transformers import DetrImageProcessor, DetrForObjectDetection
import torch import torch
@ -38,4 +24,5 @@ for score, label, box in zip(results["scores"], results["labels"], results["boxe
f"Detected {model.config.id2label[label.item()]} with confidence " f"Detected {model.config.id2label[label.item()]} with confidence "
f"{round(score.item(), 3)} at location {box}" f"{round(score.item(), 3)} at location {box}"
) )
``` ```
</details>