微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

libcaffeconverter将caffe导入onnx的错误

如何解决libcaffeconverter将caffe导入onnx的错误

问题描述


我正在跟踪https://github.com/onnx/onnx-docker/blob/master/onnx-ecosystem/converter_scripts/caffe_coreml_onnx.ipynb上的caffe to onnx教程,并且遇到此错误: ImportError:无法从“ coremltools”导入名称“ libcaffeconverter”

在进一步调查中,我意识到coremltools目录中没有libcaffeconverter。

系统信息


OS平台和发行版:Windows 10 x64 ONNX版本:1.7.0 Python版本:3.7.7 Protobuf版本:3.13.0 Visual Studio版本(如果适用):2017版本15.9.28307.1234

复制说明


创建一个新环境并运行“ pip install coremltools == 4.0b3”

https://github.com/onnx/onnx-docker/blob/master/onnx-ecosystem/converter_scripts/caffe_coreml_onnx.ipynb上运行代码

OR

运行:

导入coremltools coremltools.converters.caffe.convert(无)

预期行为


我希望代码根据代码中指定的caffe文件创建并保存一个onnx文件

解决方法

TL;DR:切换到 python 3.6

像 OP 一样,问题出在 import coremltools coremltools.converters.caffe.convert(None),产生 `ImportError: cannot import name 'libcaffeconverter' 这可以通过在 python 3.6 中使用虚拟环境来避免

解决方法:

  1. 在终端中使用您选择的包管理器创建一个 python 3.6 环境: conda create -n <your env name here> python=3.6 anaconda 要么 virtualenv --python=/usr/bin/python3.6

  2. 激活虚拟环境 conda activate <your env name here>(或 pip 等价物)

  3. 导航到您的项目目录并运行您的文件 python <your script here>.py

现在它应该按计划工作

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。