[RT-Detr] - Add RT-Detr Onnx Config#2040
Conversation
|
Hi @fxmarty, @echarlaix, @JingyaHuang, @michaelbenayoun, I'm creating this PR following @qubvel and @ArthurZucker advices on that PR: huggingface/transformers#33877 |
echarlaix
left a comment
There was a problem hiding this comment.
Thanks a lot for the addition @YHallouard
|
Hi @echarlaix, thanks for the review! I added "PekingU/rtdetr_r18vd" for testing, but I saw that you used hf-internal-testing. Should I create a random initiated model their ? |
|
Hi @echarlaix, what are your insights ? I made the changes but should I create a random initiated model in hf-internal-testing ? |
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
|
Apologies for the delay @YHallouard, just fixed the conflicts so that we can soon merge. Two things must be fixed before we can merge :
|
| def inputs(self) -> Dict[str, Dict[int, str]]: | ||
| return { | ||
| "pixel_values": {0: "batch_size", 1: "num_channels", 2: "height", 3: "width"}, | ||
| "pixel_mask": {0: "batch_size"}, |
There was a problem hiding this comment.
why is it needed ? (not needed for ViT models but might be needed for RT-DETR)
There was a problem hiding this comment.
also if needed shouldn't the height / width dimension set to dynamic as well?
|
Looks like this task got stuck, do you need any help on it? Would be really great to have RT-DETR exporter for onnx. |
|
Closed with #2201 |
What does this PR do?
Fixes # (issue)
Before submitting
Who can review?