diff --git a/README.md b/README.md index 2b62af2..6ab8a1c 100644 --- a/README.md +++ b/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}" ) -``` \ No newline at end of file +``` + \ No newline at end of file