问题描述
问题描述
我正在跟踪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”
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 中使用虚拟环境来避免
解决方法:
-
在终端中使用您选择的包管理器创建一个 python 3.6 环境:
conda create -n <your env name here> python=3.6 anaconda
要么virtualenv --python=/usr/bin/python3.6
-
激活虚拟环境
conda activate <your env name here>
(或 pip 等价物) -
导航到您的项目目录并运行您的文件
python <your script here>.py
现在它应该按计划工作