Detectron: Is it possible to export Detectron pre-trained models as onnx models?

Created on 13 Feb 2018 · 6Comments · Source: facebookresearch/Detectron

Is it possible to export Detectron pre-trained models as onnx models?
If it is not supported now, Do you have plans to support this in the near future?

planned feature

Source

kumardesappan

Most helpful comment

The issue is closed but I can't seem to find ONNX models for detectron anywhere - is there still plan for that ?

borisfom on 9 Mar 2018

👍14

All 6 comments

alaaadnan229 on 14 Feb 2018

cc @dzhulgakov and @orionr

ir413 on 14 Feb 2018

Hi @kumardesappan. We are definitely interested in exporting Detectron models to ONNX. However, there is work to be done there. In fact, we also have work to do in exporting Detectron models to Caffe2 pb files, which can be run with base Caffe2 and C++ (on a mobile device, for example).

We're in progress on adding some Detectron models to the Caffe2 model zoo at

https://github.com/caffe2/models

and eventually some of those will make their way into the ONNX model zoo at

https://github.com/onnx/models

No ETA yet, but something we are working on. Hope that helps!