Export ONNX model
williamhyin opened this issue · comments
Hello
Could you share your onnx export script, which can help us a lot.
Thanks!
Hello @williamhyin good point. Currently, I don't have a bandwidth for doing this, but I will keep this in my backlog.
Hello @williamhyin good point. Currently, I don't have a bandwidth for doing this, but I will keep this in my backlog.
Hi ,
Thanks for your reply. II noticed that DD3D uses the detectron2 architecture. Detectron2-onnx conversation is based on Caffe2Tracel. Is it possible to use the deployment API of detectron2 to perform ONNX conversion?
link: https://detectron2.readthedocs.io/en/latest/_modules/detectron2/export/api.html#Caffe2Model
tutorial: https://detectron2.readthedocs.io/en/latest/tutorials/deployment.html
But I faced the problem when i use onnx-export command:
onnxModel = export_onnx_model(cfg, model, inputs)
onnx.save(onnxModel, "/content/deploy.onnx")
bug: assert isinstance(cfg, CfgNode), cfg
It seems like the dd3d-cfg is not the official format cfg file.
And the inputs from dataloder is more complex.
Could you give me some idea or suggestion?
Thanks! I look forward to hearing from you.
Hello @williamhyin good point. Currently, I don't have a bandwidth for doing this, but I will keep this in my backlog.
Hi ,
Thanks for your reply. II noticed that DD3D uses the detectron2 architecture. Detectron2-onnx conversation is based on Caffe2Tracel. Is it possible to use the deployment API of detectron2 to perform ONNX conversion?
link: https://detectron2.readthedocs.io/en/latest/_modules/detectron2/export/api.html#Caffe2Model tutorial: https://detectron2.readthedocs.io/en/latest/tutorials/deployment.html
But I faced the problem when i use onnx-export command:
onnxModel = export_onnx_model(cfg, model, inputs) onnx.save(onnxModel, "/content/deploy.onnx")
bug: assert isinstance(cfg, CfgNode), cfg
It seems like the dd3d-cfg is not the official format cfg file. And the inputs from dataloder is more complex. Could you give me some idea or suggestion?
Thanks! I look forward to hearing from you.
Have you saved you problem?