mirrors
/
RE-OWOD
mirror of https://github.com/RE-OWOD/RE-OWOD
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
RE-OWOD/detectron2/export
History
RE-OWOD 665719112e
Add files via upload 		2022-01-04 13:35:52 +08:00
..
README.md Add files via upload 2022-01-04 13:35:52 +08:00
__init__.py Add files via upload 2022-01-04 13:35:52 +08:00
api.py Add files via upload 2022-01-04 13:35:52 +08:00
c10.py Add files via upload 2022-01-04 13:35:52 +08:00
caffe2_export.py Add files via upload 2022-01-04 13:35:52 +08:00
caffe2_inference.py Add files via upload 2022-01-04 13:35:52 +08:00
caffe2_modeling.py Add files via upload 2022-01-04 13:35:52 +08:00
patcher.py Add files via upload 2022-01-04 13:35:52 +08:00
shared.py Add files via upload 2022-01-04 13:35:52 +08:00
torchscript.py Add files via upload 2022-01-04 13:35:52 +08:00

README.md

This directory contains code to prepare a detectron2 model for deployment. Currently it supports exporting a detectron2 model to Caffe2 format through ONNX.

Please see documentation for its usage.

Acknowledgements

Thanks to Mobile Vision team at Facebook for developing the Caffe2 conversion tools.

Thanks to Computing Platform Department - PAI team at Alibaba Group (@bddpqq, @chenbohua3) who help export Detectron2 models to TorchScript.