search

IDEA-Research / DINO

[ICLR 2023] Official implementation of the paper "DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detection"
Apache License 2.0
2.15k stars 232 forks source link

DINO onnx conversion #150

Open Simardeep27 opened 1 year ago

Simardeep27 commented 1 year ago

Hi,

I am currently working on ONNX conversion for DINO model. The MSDeformAttention function is currently implemented only on GPU which causes error while conversion of the model. Is there any workaround for this?

FengLi-ust commented 1 year ago

We did not try onnx before, so currently, we could not offer any suggestions.

Shuai-Xie commented 1 year ago

@Simardeep27 @FengLi-ust ME too.

I Gusess OpenMMLab mmdeploy maybe a walk around as it supports DETR and DeformableDETR. And DINO is integrating now. https://github.com/open-mmlab/mmdetection/pull/9149