TahaDouaji
commited on
Commit
·
7207383
1
Parent(s):
e05068a
Update README.md
Browse files
README.md
CHANGED
@@ -1,17 +1,3 @@
|
|
1 |
-
---
|
2 |
-
tags:
|
3 |
-
- object-detection
|
4 |
-
---
|
5 |
-
|
6 |
-
|
7 |
-
## Model description
|
8 |
-
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)
|
9 |
-
|
10 |
-
## Training data
|
11 |
-
The model was trained on ICDAR2019 Table Dataset
|
12 |
-
|
13 |
-
### How to use
|
14 |
-
|
15 |
```python
|
16 |
from transformers import DetrImageProcessor, DetrForObjectDetection
|
17 |
import torch
|
@@ -38,4 +24,5 @@ for score, label, box in zip(results["scores"], results["labels"], results["boxe
|
|
38 |
f"Detected {model.config.id2label[label.item()]} with confidence "
|
39 |
f"{round(score.item(), 3)} at location {box}"
|
40 |
)
|
41 |
-
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
```python
|
2 |
from transformers import DetrImageProcessor, DetrForObjectDetection
|
3 |
import torch
|
|
|
24 |
f"Detected {model.config.id2label[label.item()]} with confidence "
|
25 |
f"{round(score.item(), 3)} at location {box}"
|
26 |
)
|
27 |
+
```
|
28 |
+
</details>
|