optimum icon indicating copy to clipboard operation
optimum copied to clipboard

Add ONNX export support for RT-DETR models

Open xenova opened this issue 1 year ago • 5 comments
trafficstars

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?

xenova avatar Jun 28 '24 13:06 xenova

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?

fxmarty avatar Jun 28 '24 15:06 fxmarty

Done! @fxmarty :) I also set CHW axes to be static (must be what the model was trained on)

xenova avatar Jun 28 '24 15:06 xenova

static shapes might not be necessary actually: https://github.com/huggingface/transformers/pull/31640

xenova avatar Jun 28 '24 16:06 xenova

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!

MattLiutt avatar Aug 16 '24 02:08 MattLiutt

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.

github-actions[bot] avatar Dec 06 '24 02:12 github-actions[bot]