Skip to content

[Feat] Add Layout model LW-DETR#2059

Merged
felixdittrich92 merged 15 commits into
mindee:mainfrom
felixdittrich92:lw-detr
May 12, 2026
Merged

[Feat] Add Layout model LW-DETR#2059
felixdittrich92 merged 15 commits into
mindee:mainfrom
felixdittrich92:lw-detr

Conversation

@felixdittrich92
Copy link
Copy Markdown
Collaborator

@felixdittrich92 felixdittrich92 commented May 6, 2026

This PR:

  • Add LW-DETR layout detection model
  • Add corresponding tests

@felixdittrich92 felixdittrich92 added this to the 1.1.0 milestone May 6, 2026
@felixdittrich92 felixdittrich92 self-assigned this May 6, 2026
@felixdittrich92 felixdittrich92 added topic: documentation Improvements or additions to documentation module: models Related to doctr.models ext: tests Related to tests folder module: transforms Related to doctr.transforms framework: pytorch Related to PyTorch backend type: new feature New feature ext: docs Related to docs folder topic: layout detection Related to document layout detection labels May 6, 2026
@felixdittrich92
Copy link
Copy Markdown
Collaborator Author

felixdittrich92 commented May 6, 2026

@simont2k A first draft not yet done - If you want it's ready to have a first look - I will split the PR into several PRs to make the review easier

TODOs:

  • Fix mypy
  • Further clean-up / minor improvements
  • Add missing documentation

@codecov
Copy link
Copy Markdown

codecov Bot commented May 7, 2026

Codecov Report

❌ Patch coverage is 97.54768% with 18 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.71%. Comparing base (ec4e076) to head (fc8fb63).

Files with missing lines Patch % Lines
doctr/models/layout/lw_detr/pytorch.py 97.79% 7 Missing ⚠️
doctr/models/layout/lw_detr/base.py 96.15% 4 Missing ⚠️
doctr/models/layout/lw_detr/layers/pytorch.py 98.39% 4 Missing ⚠️
doctr/models/layout/zoo.py 92.00% 2 Missing ⚠️
doctr/models/layout/predictor/pytorch.py 96.66% 1 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2059      +/-   ##
==========================================
+ Coverage   96.43%   96.71%   +0.27%     
==========================================
  Files         147      156       +9     
  Lines        6486     7220     +734     
==========================================
+ Hits         6255     6983     +728     
- Misses        231      237       +6     
Flag Coverage Δ
unittests 96.71% <97.54%> (+0.27%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@felixdittrich92 felixdittrich92 marked this pull request as ready for review May 12, 2026 10:25
@felixdittrich92 felixdittrich92 merged commit 1dcc78f into mindee:main May 12, 2026
49 of 51 checks passed
@felixdittrich92 felixdittrich92 deleted the lw-detr branch May 12, 2026 18:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ext: docs Related to docs folder ext: tests Related to tests folder framework: pytorch Related to PyTorch backend module: models Related to doctr.models module: transforms Related to doctr.transforms topic: documentation Improvements or additions to documentation topic: layout detection Related to document layout detection type: new feature New feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants