mmdeploy自定义算子

发表于 更新于

如果mmdeploy是从源码安装的话，在导出模型时可能会出现自定义算子报错的情况。
此时需要编译自定义算子的动态库：(见https://github.com/open-mmlab/mmdeploy/blob/dev-1.x/docs/zh_cn/01-how-to-build/build_from_source.md)

此时能够正常导出onnx模型。

在使用onnx进行推理时，代码中需要注册自定义算子(详见https://mmcv.readthedocs.io/zh-cn/v1.3.15/deployment/onnxruntime_op.html)

1
2
3
4
5
ops_lib = "..." # 自定义算子库文件位置

session_options = ort.SessionOptions()
session_options.register_custom_ops_library(ops_lib) # 注册自定义算子
...