我想把一个模型转换成onnx 但是当我导入 caffe2_export 以将模型导出到 onnx 时,我得到了这个错误
from detectron2.export import caffe2_export
错误 导入 onnx.optimizer ModuleNotFoundError:没有名为“onnx.optimizer”的模块
是因为从onnx 1.9版本开始去掉了onnxoptimizer
如果确实需要使用,可以安装onnx 1.8版本。
pip 安装 onnx==1.8.1
注意:我不推荐这种方法。
如果你真的想将模型导出为 onnx 格式,我建议你使用 export_model.py.
对于遇到此问题的任何其他人来说,将 Detectron2 模型导出为 ONNX 格式一点都不简单。导出包括需要 Caffe2 的操作,然后需要替换这些操作。
有人在这里管理它,但我无法复制它https://github.com/NVIDIA/TensorRT/issues/2546