Export to ONNX

[crey21]

Hi,

Is it possible to export the model used in the demo.py to ONNX ?

I've tried with the official Python methods (export and dynamo_export) with no luck, and the detectron2 methods that rely on caffe2 seem out of touch (I'd rather avoid compile my own pytorch with caffe2 enabled except if it's the only way).

Has anyone done it, or is it documented somewhere ?

(My root problem is that I cannot get it to run on linux-aarch64, even with config.MODEL.DEVICE = "cpu", whereas the same code and dependencies versions run fine on macos-arm64.)

Thanks

[retazo0018]

Hi @mlzxy,

I have the same issue as well. I am not able to export the model to onnx. Please help if anyone has tried something similar.

Thanks