Update README.md
This commit is contained in:
parent
e05068a73a
commit
7207383528
17
README.md
17
README.md
|
@ -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>
|
Loading…
Reference in New Issue