aoiandroid's picture
Document ONNX sibling repo and TranslateBlue usage
9b33f2d verified
|
Raw
History Blame Contribute Delete
1.19 kB
# DocLayout-YOLO imgsz1280 (DocStructBench)
| Artifact | Hugging Face |
|----------|----------------|
| PyTorch | [aoiandroid/DocLayout-YOLO-DocStructBench-imgsz1280-2501](https://huggingface.co/aoiandroid/DocLayout-YOLO-DocStructBench-imgsz1280-2501) |
| ONNX (1280²) | [aoiandroid/doclayout-yolo-docstructbench-imgsz1280-onnx](https://huggingface.co/aoiandroid/doclayout-yolo-docstructbench-imgsz1280-onnx) |
TranslateBlue resolves the active variant via `DocLayoutModelResolver`: when `doclayout_yolo_docstructbench_imgsz1280_2501.onnx` is present under `models/`, `Documents/DocLayoutYOLO/`, or after Hub download, PDF and image layout use **1280×1280** input. Otherwise the app falls back to **imgsz1024** (Core ML precompute + ONNX).
Re-export ONNX from the `.pt` checkpoint:
```bash
python3 Scripts/export_doclayout_imgsz1280_onnx.py
hf upload aoiandroid/doclayout-yolo-docstructbench-imgsz1280-onnx \
models/doclayout_yolo_docstructbench_imgsz1280_2501.onnx \
doclayout_yolo_docstructbench_imgsz1280_2501.onnx
```
Upstream duplicate: [juliozhao/DocLayout-YOLO-DocStructBench-imgsz1280-2501](https://huggingface.co/juliozhao/DocLayout-YOLO-DocStructBench-imgsz1280-2501).