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
315 B
Markdown
20 lines
315 B
Markdown
## MMCV中ONNX模块简介 (实验性)
|
||
|
||
### register_extra_symbolics
|
||
|
||
在将PyTorch模型导出成ONNX时，需要注册额外的符号函数
|
||
|
||
#### 范例
|
||
|
||
```python
|
||
import mmcv
|
||
from mmcv.onnx import register_extra_symbolics
|
||
|
||
opset_version = 11
|
||
register_extra_symbolics(opset_version)
|
||
```
|
||
|
||
#### 常见问题
|
||
|
||
- 无
|