Skip to content

feat: add batch prediction for layout predictor #107

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 3 commits into
base: main
Choose a base branch
from

Conversation

SteffiVanHees
Copy link

Add batch prediction for layout predictor.

Checklist:

  • Documentation has been updated, if necessary.
  • Examples have been added, if necessary.
  • Tests have been added, if necessary.

Copy link

mergify bot commented May 26, 2025

Merge Protections

Your pull request matches the following merge protections and will not be merged until they are valid.

🟢 Enforce conventional commit

Wonderful, this rule succeeded.

Make sure that we follow https://www.conventionalcommits.org/en/v1.0.0/

  • title ~= ^(fix|feat|docs|style|refactor|perf|test|build|ci|chore|revert)(?:\(.+\))?(!)?:

@SteffiVanHees SteffiVanHees force-pushed the layoutpredict_batch branch from 2157f13 to aa2e39b Compare May 26, 2025 13:52
@nikos-livathinos
Copy link
Collaborator

@SteffiVanHees thank you for your interest to support batch layout predictions.

How do you plan to use the LayoutPredictor:predict_batch() method? Do you plan to make changes in Docling that can take advantage of the batch processing?

Also, it would be nice to extend the unit tests for the LayoutPredictor and the demo_layout_predictor CLI app.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants