Upload folder using huggingface_hub
Browse files- README.md +71 -0
- rtdetr/model.onnx +3 -0
- slanet-plus/model.onnx +3 -0
README.md
ADDED
|
@@ -0,0 +1,71 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- document-layout-analysis
|
| 5 |
+
- table-structure-recognition
|
| 6 |
+
- onnx
|
| 7 |
+
- kreuzberg
|
| 8 |
+
---
|
| 9 |
+
|
| 10 |
+
# Kreuzberg Layout Models
|
| 11 |
+
|
| 12 |
+
ONNX models used by [Kreuzberg](https://kreuzberg.dev) for document layout detection and table structure recognition.
|
| 13 |
+
|
| 14 |
+
## Models
|
| 15 |
+
|
| 16 |
+
### RT-DETR (Document Layout Detection)
|
| 17 |
+
|
| 18 |
+
| Property | Value |
|
| 19 |
+
|----------|-------|
|
| 20 |
+
| **Path** | `rtdetr/model.onnx` |
|
| 21 |
+
| **Size** | 169 MB |
|
| 22 |
+
| **Precision** | FP32 |
|
| 23 |
+
| **Architecture** | RT-DETR v2 (Real-Time Detection Transformer) |
|
| 24 |
+
| **Input** | `images`: `[batch, 3, 640, 640]` f32 (ImageNet-normalized, letterboxed) |
|
| 25 |
+
| **Input** | `orig_target_sizes`: `[batch, 2]` i64 (original `[height, width]`) |
|
| 26 |
+
| **Outputs** | `labels` i64, `boxes` f32 `[batch, N, 4]`, `scores` f32 |
|
| 27 |
+
| **Classes** | 17 document layout classes |
|
| 28 |
+
| **SHA256** | `3bf2fb0ee6df87435b7ae47f0f3930ec3dc97ec56fd824acc6d57bc7a6b89ef2` |
|
| 29 |
+
|
| 30 |
+
**Layout Classes:** Caption, Footnote, Formula, ListItem, PageFooter, PageHeader, Picture, SectionHeader, Table, Text, Title, DocumentIndex, Code, CheckboxSelected, CheckboxUnselected, Form, KeyValueRegion
|
| 31 |
+
|
| 32 |
+
### SLANet-plus (Table Structure Recognition)
|
| 33 |
+
|
| 34 |
+
| Property | Value |
|
| 35 |
+
|----------|-------|
|
| 36 |
+
| **Path** | `slanet-plus/model.onnx` |
|
| 37 |
+
| **Size** | 7.8 MB |
|
| 38 |
+
| **Precision** | FP32 |
|
| 39 |
+
| **Architecture** | SLANet-plus (Sequence-to-Sequence table decoder) |
|
| 40 |
+
| **Input** | `x`: `[1, 3, 488, 488]` f32 (BGR channel order, ImageNet-normalized) |
|
| 41 |
+
| **Outputs** | `[1, seq_len, 8]` cell bbox corners, `[1, seq_len, 50]` HTML token probabilities |
|
| 42 |
+
| **Vocabulary** | 50 tokens (HTML structure tags, rowspan/colspan 1-20, sos/eos) |
|
| 43 |
+
| **SHA256** | `e0bff8da087f9b83629f1e1a6e0f8252fc2de85a7d80415b3510fc521338da3d` |
|
| 44 |
+
|
| 45 |
+
## Attribution & Provenance
|
| 46 |
+
|
| 47 |
+
### RT-DETR
|
| 48 |
+
|
| 49 |
+
This model is mirrored from [docling-project/docling-layout-heron-onnx](https://huggingface.co/docling-project/docling-layout-heron-onnx), created by the [Docling](https://github.com/docling-project/docling) team at IBM Research.
|
| 50 |
+
|
| 51 |
+
- **Original repository:** [docling-project/docling-layout-heron-onnx](https://huggingface.co/docling-project/docling-layout-heron-onnx)
|
| 52 |
+
- **License:** Apache-2.0
|
| 53 |
+
- **Architecture paper:** Zhao et al., "DETRs Beat YOLOs on Real-time Object Detection" ([arXiv:2304.08069](https://arxiv.org/abs/2304.08069))
|
| 54 |
+
- **Training data:** DocLayNet and internal IBM document datasets
|
| 55 |
+
|
| 56 |
+
### SLANet-plus
|
| 57 |
+
|
| 58 |
+
This model was converted from PaddlePaddle format to ONNX using [Paddle2ONNX](https://github.com/PaddlePaddle/Paddle2ONNX). The original model is from the [PaddleOCR](https://github.com/PaddlePaddle/PaddleOCR) project by PaddlePaddle.
|
| 59 |
+
|
| 60 |
+
- **Original repository:** [PaddlePaddle/SLANet_plus](https://huggingface.co/PaddlePaddle/SLANet_plus)
|
| 61 |
+
- **License:** Apache-2.0
|
| 62 |
+
- **Architecture paper:** "PP-StructureV2: A Stronger Document Analysis System" ([arXiv:2210.05391](https://arxiv.org/abs/2210.05391))
|
| 63 |
+
- **Conversion:** PaddlePaddle inference format → ONNX via Paddle2ONNX (opset 17)
|
| 64 |
+
|
| 65 |
+
## Usage
|
| 66 |
+
|
| 67 |
+
These models are automatically downloaded and cached by the [Kreuzberg](https://kreuzberg.dev) document extraction library. See the [layout extraction documentation](https://kreuzberg.dev) for details.
|
| 68 |
+
|
| 69 |
+
## License
|
| 70 |
+
|
| 71 |
+
All models in this repository are distributed under the [Apache-2.0 License](https://www.apache.org/licenses/LICENSE-2.0), consistent with the licenses of the original models.
|
rtdetr/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3bf2fb0ee6df87435b7ae47f0f3930ec3dc97ec56fd824acc6d57bc7a6b89ef2
|
| 3 |
+
size 169089059
|
slanet-plus/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e0bff8da087f9b83629f1e1a6e0f8252fc2de85a7d80415b3510fc521338da3d
|
| 3 |
+
size 7781255
|