Update README.md
This commit is contained in:
parent
e05068a73a
commit
7207383528
15
README.md
15
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
|
||||
|
@ -39,3 +25,4 @@ for score, label, box in zip(results["scores"], results["labels"], results["boxe
|
|||
f"{round(score.item(), 3)} at location {box}"
|
||||
)
|
||||
```
|
||||
</details>
|
Loading…
Reference in New Issue