mirror of
https://github.com/open-mmlab/mmcv.git
synced 2025-06-03 21:54:52 +08:00
2b39d7a8ec
* 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>
20 lines
344 B
Markdown
20 lines
344 B
Markdown
## 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
|
|
|
|
```python
|
|
import mmcv
|
|
from mmcv.onnx import register_extra_symbolics
|
|
|
|
opset_version = 11
|
|
register_extra_symbolics(opset_version)
|
|
```
|
|
|
|
#### FAQs
|
|
|
|
- None
|