Add ONNX config for RT-DETR (and RT-DETRv2) #2201
echarlaix
merged
10 commits into
huggingface:main
from
qubvel:add-onnx-export-rt-detr
Add RTDetr Onnx Config
e92cabe3
add test and comment
6c8106c8
fix style
8c04a0ba
Update tests large model
e35483c3
fix expected configs model type
30d1c94f
Fix inputs shape and remove pixel_mask
9a4a4e3d
Add rt-detr-v2
07a13684
Add tests
3ece80f0
Add separate V2 config
2e0290b3
echarlaix
approved these changes
on 2025-03-03
Update min size according to num_queries
04c3ec3c
Login to write a write a comment.
Login via GitHub