Skip to content

[Feat] Add Vit-Det as LW-DETR encoder#2063

Merged
felixdittrich92 merged 3 commits into
mindee:mainfrom
felixdittrich92:vit-det
May 11, 2026
Merged

[Feat] Add Vit-Det as LW-DETR encoder#2063
felixdittrich92 merged 3 commits into
mindee:mainfrom
felixdittrich92:vit-det

Conversation

@felixdittrich92
Copy link
Copy Markdown
Collaborator

@felixdittrich92 felixdittrich92 commented May 7, 2026

This PR:

  • Add Vit-Det as classification model (encoder for LW-DETR)
  • Add corresponding tests
  • Add docs entry
  • Add preps for layout models (onnx export / device and dtype set)
  • Add ChannelLayerNorm which will be required by LW-DETR

@felixdittrich92 felixdittrich92 added this to the 1.1.0 milestone May 7, 2026
@felixdittrich92 felixdittrich92 self-assigned this May 7, 2026
@felixdittrich92 felixdittrich92 added topic: documentation Improvements or additions to documentation module: models Related to doctr.models module: utils Related to doctr.utils framework: pytorch Related to PyTorch backend topic: character classification Related to the task of character classification topic: onnx ONNX-related type: new feature New feature ext: docs Related to docs folder labels May 7, 2026
@felixdittrich92
Copy link
Copy Markdown
Collaborator Author

Related to #2059

@felixdittrich92 felixdittrich92 requested a review from simont2k May 7, 2026 09:51
@codecov
Copy link
Copy Markdown

codecov Bot commented May 7, 2026

Codecov Report

❌ Patch coverage is 94.90741% with 11 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.43%. Comparing base (db9d92a) to head (aa68b04).

Files with missing lines Patch % Lines
doctr/models/modules/layers/pytorch.py 28.57% 10 Missing ⚠️
...tr/models/classification/vit_det/layers/pytorch.py 98.68% 1 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2063      +/-   ##
==========================================
- Coverage   96.48%   96.43%   -0.05%     
==========================================
  Files         142      146       +4     
  Lines        6111     6319     +208     
==========================================
+ Hits         5896     6094     +198     
- Misses        215      225      +10     
Flag Coverage Δ
unittests 96.43% <94.90%> (-0.05%) ⬇️

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 merged commit 53c58dc into mindee:main May 11, 2026
42 of 45 checks passed
@felixdittrich92 felixdittrich92 deleted the vit-det branch May 11, 2026 08:49
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 framework: pytorch Related to PyTorch backend module: models Related to doctr.models module: utils Related to doctr.utils topic: character classification Related to the task of character classification topic: documentation Improvements or additions to documentation topic: onnx ONNX-related type: new feature New feature

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant