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

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
from transformers import DetrImageProcessor, DetrForObjectDetection
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"{round(score.item(), 3)} at location {box}"
)
```
```
</details>