Upload folder using huggingface_hub

README.md

@@ -0,0 +1,45 @@
+---
+license: apache-2.0
+tags:
+  - document-layout
+  - object-detection
+  - yolo
+library_name: ultralytics
+---
+
+# DocLayout-YOLO - Docstructbench
+
+Document layout detection model. Paper: [DocLayout-YOLO](https://arxiv.org/abs/2410.12628)
+
+## Classes (10)
+
+- `title`
+- `plain_text`
+- `abandon`
+- `figure`
+- `figure_caption`
+- `table`
+- `table_caption`
+- `table_footnote`
+- `isolate_formula`
+- `formula_caption`
+
+## Usage
+
+```python
+pip install anyformat-doclayout
+```
+
+```python
+from anyformat.doclayout import DocLayoutModel, download_converted
+
+# Download weights from this repo
+weights = download_converted("docstructbench")
+
+# Run inference
+model = DocLayoutModel(weights)
+results = model.predict("document.png")
+
+for det in results:
+    print(f"{det['class_name']}: {det['confidence']:.2f}")
+```

config.json

@@ -0,0 +1,19 @@
+{
+  "model_type": "doclayout-yolo",
+  "model_name": "docstructbench",
+  "architecture": "yolov10m-g2l-crm",
+  "num_classes": 10,
+  "class_names": [
+    "title",
+    "plain_text",
+    "abandon",
+    "figure",
+    "figure_caption",
+    "table",
+    "table_caption",
+    "table_footnote",
+    "isolate_formula",
+    "formula_caption"
+  ],
+  "input_size": 1024
+}

model.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5817d783e3f7b5854a06b5d4b96f649257315f985892331705d5f596113b236
+size 75725584

model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:603d42c9ab6822b1aacde18a2615a6a3d415ec8eb269d2ac3bc6504d7472e69c
+size 81028589