DINO icon indicating copy to clipboard operation
DINO copied to clipboard

Plan to support for ONNX

Open jhj7905 opened this issue 2 years ago • 4 comments

These days, I trained and evaluated my own dataset and got significant result with your model compared to YOLOS. Now, I have tried to convert DINO(pytorch) into DINO(onnx or tensorrt). But there are a few reference. so still in trouble. Do you have a plan to support for onnx? To all DINO contributors, thank you for sharing the code!

jhj7905 avatar Aug 24 '22 11:08 jhj7905

there were some exploration in converting DETR to onnx, hope it help. https://github.com/jinfagang/yolov7_d2 Haven't try DINO though, but am trying porting DINO to detectron2 these days, converting transformer based model to onnx a little bit cubersome, even you can export, probabely not work on TensorRT, but we have made it work on DETR already.

lucasjinreal avatar Aug 24 '22 11:08 lucasjinreal

@jinfagang Thank you for sharing the reference. i will try to apply it to convert DETR(pytorch) to onnx(or tenssorrt)

jhj7905 avatar Aug 25 '22 01:08 jhj7905

@jinfagang Thank you for sharing the reference. i will try to apply it to convert DETR(pytorch) to onnx(or tenssorrt)

Is your conversion successful?

yangzheng0218 avatar Mar 15 '23 08:03 yangzheng0218

Did you successfully converted DINO to ONNX? thank you

lonalala avatar Nov 30 '23 10:11 lonalala