mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity
mmcv/docs/deployment/onnx.md
q.yao 2b39d7a8ec
[Docs] Add zh_cn document of ONNX (#1331) 
* add doc-cn of ONNX

* Update docs_zh_CN/deployment/onnxruntime_custom_ops.md

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Apply suggestions from code review

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* update doc of cummax

* fix en doc of softnms

* update heading

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2021-09-24 16:57:20 +08:00

344 B
Raw Blame History

Introduction of onnx module in MMCV (Experimental)

register_extra_symbolics

Some extra symbolic functions need to be registered before exporting PyTorch model to ONNX.

Example

import mmcv
from mmcv.onnx import register_extra_symbolics

opset_version = 11
register_extra_symbolics(opset_version)

FAQs

  • None