pytorch 探测器2至Onyx

5ktev3wc  于 2023-02-08  发布在  其他
关注(0)|答案(1)|浏览(143)

我想将模型转换为onnx,但当我导入caffe2_export以将模型导出到onnx时,我收到此错误

from detectron2.export import caffe2_export

导入onnx时出错。优化器模块未找到错误:没有名为“onnx.optimizer”的模块

j9per5c4

j9per5c41#

这是因为onnxoptimizer从onnx版本1.9起就被删除了。
如果你真的需要使用它,你可以调用install onnx version 1.8。

pip安装在nx上== 1.8.1

    • 注**:我不推荐这种方法。

相关问题