optimum
optimum copied to clipboard
Add ONNX export support for RT-DETR models
What does this PR do?
cc @merveenoyan
Fixes # (issue)
Before submitting
- [ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
- [ ] Did you make sure to update the documentation with your changes?
- [ ] Did you write any new necessary tests?
Who can review?
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.
@xenova can you add a test?
Done! @fxmarty :) I also set CHW axes to be static (must be what the model was trained on)
static shapes might not be necessary actually: https://github.com/huggingface/transformers/pull/31640
Hi, may I know that if it's possible to run with onnxruntime instead of using optimum and obtaining model processor using HF pipeline? Also, I tried to export with Cuda but it seems not working. Is there anything I can tweak? Thanks again for the great work!
This PR has been marked as stale because it has been open for 90 days with no activity. This thread will be automatically closed in 30 days if no further activity occurs.