core-ocr-and-kie
Run docTR end-to-end OCR and KIE inference with predictor APIs, rotation/layout/table options, batching, devices, and output checks.
- License 许可
- Apache-2.0
- 最近更新
- 2026-09-03
- Stars
- ★ 351
- Forks
- 27
01 前言 Front Matter
- name
- core-ocr-and-kie
- license
- Apache 2.0
- metadata
- { "disco-role": "operating" }
- description
- Run docTR end-to-end OCR and KIE inference with predictor APIs, rotation/layout/table options, batching, devices, and output checks.
- disable-model-invocation
- true
02 资源文件 3
- 参考文档references/ocr-kie-workflows.md12.7 KB
- 参考文档references/troubleshooting.md9.2 KB
- 脚本scripts/ocr_api_smoke.py8.9 KB
03 SKILL.md 正文
core-ocr-and-kie
Use this sub-skill when the task is to run or debug docTR's Python end-to-end inference pipelines:
doctr.io.DocumentFileor a list of RGB NumPy pages ->doctr.models.ocr_predictor(...)->Document.doctr.io.DocumentFileor a list of RGB NumPy pages ->doctr.models.kie_predictor(...)->KIEDocument.- Predictor-level flags for rotated/skewed pages, orientation/language detection, layout detection, table detection, ignored layout regions, batching, and device movement.
- Sanity-checking output object shapes, page counts, geometry conventions, and the no-download/random-weight caveat.
Do not use this sub-skill for raw file loading/export format details, model zoo catalogs/custom weights/ONNX export, CLI wrappers, training, datasets, or evaluation scripts. Route those tasks to sibling sub-skills:
- Document loading and result export/render details: document-io-and-exports
- Architecture catalogs, custom models, optimization, export, vocab whitelisting: models-and-customization
doctr-cliand bundled command-line helpers: cli-and-scripts- Datasets, training, metrics, and evaluation: datasets-training-and-evaluation
Read first
- For API recipes and option interactions, read references/ocr-kie-workflows.md.
- For failure diagnosis, read references/troubleshooting.md.
- For a safe import/factory/optional-forward smoke check, run or adapt scripts/ocr_api_smoke.py.
Operating rules
- Prefer explicit
det_arch,reco_arch,pretrained,pretrained_backbone,assume_straight_pages, and batch-size choices when reproducibility matters. - Treat
pretrained=Falseas an API/shape smoke mode only; randomly initialized detection and recognition models do not produce useful OCR. - Be explicit about network/cache behavior: pretrained model weights and some orientation helpers may download on first use unless already cached or disabled.
- Validate outputs before downstream use:
Document.pagesfor OCR,KIEDocument.pages[*].predictionsfor KIE, optionalpage.layout, optionalpage.tables, and relative geometries.
04 同仓库更多 Skills
passnet-triton-opt
Make a single PassNet Triton kernel fast AND numerically faithful: performance model (when a replacement can win at all), block/grid/warp tuning, autotune policy, launch overhead, and per-op numeric recipes to pass the dtype baseline tolerances. Use when a pass matches and is correct but speedup below expected, or when correctness fails by small numeric margins.
passnet-solve
END-TO-END playbook for solving one PassNet sample: analyze the computation graphs, decide the optimization strategy (what to fuse, what to replace, what to leave alone), drive the iteration loop, and maximize the sample score. This is the ENTRY skill — invoke it first for any PassNet optimization task; it tells you when to use passnet-pattern-fusion, passnet-triton-opt and passnet-feedback.
passnet-skill
PassNet GPU kernel optimization via compiler passes. Design and implement Triton-based optimization passes, create pass files under ./pass_dir/, self-evaluate with pass_evaluator, and iterate to maximize GPU speedup.
passnet-pattern-fusion
Author PassNet pass files: write patterns that actually MATCH the FX graph, pick fusion regions (which ops to absorb into one Triton kernel), and structure multi-pass submissions with the mandatory shared-dispatch architecture. Use when creating or fixing pattern()/replacement_args()/replacement_func() files, when a pass "failed to match", or when deciding how to fuse multiple kernels/ops.
passnet-orchestrate
Round/batch planning and decision support for PassNet graph optimization. Use for multi-sample triage, worker allocation, eval-budget policy, and post-eval keep/revert decisions. For solving one concrete sample end to end, use passnet-solve as the entry skill; consult this skill only when you need broader planning or a specific decision gate.
passnet-feedback
Fast, deterministic feedback for PassNet work: pre-flight pattern/match verification WITHOUT burning a GPU evaluation, per-node bottleneck analysis of a sample's graphs, and eval-log parsing into per-variant status + estimated score + failure classification. Use BEFORE every GPU evaluation (check_pattern), at the START of a sample (analyze_graph), and AFTER every evaluation (parse_eval_log).