naamanhirschfeld commited on
Commit
b46963f
·
verified ·
1 Parent(s): da3b833

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +71 -0
  2. rtdetr/model.onnx +3 -0
  3. 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