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_zh_CN/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

315 B
Raw Blame History

MMCV中ONNX模块简介 (实验性)

register_extra_symbolics

在将PyTorch模型导出成ONNX时，需要注册额外的符号函数

范例

import mmcv
from mmcv.onnx import register_extra_symbolics

opset_version = 11
register_extra_symbolics(opset_version)

常见问题

  • 无