mmcv/docs/zh_cn/deployment/onnx.md

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)

常见问题