caffe到onnx的libcaffeconverter导入错误

2024-04-26 11:44:25 发布

您现在位置:Python中文网/ 问答频道 /正文

问题描述


我在https://github.com/onnx/onnx-docker/blob/master/onnx-ecosystem/converter_scripts/caffe_coreml_onnx.ipynb遵循caffe到onnx教程,遇到以下错误: ImportError:无法从“coremltools”导入名称“libcafeconverter”

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

系统信息


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

复制指令


创建新环境并运行“pip安装coremltools==4.0b3”

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

运行:

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

预期行为


我希望代码能够基于代码中指定的caffe文件创建并保存onnx文件


Tags: docker代码https版本githubmastercomscripts
1条回答
网友
1楼 · 发布于 2024-04-26 11:44:25

TL;DR:切换到python 3.6

与OP一样,问题在于import coremltools coremltools.converters.caffe.convert(None),产生“ImportError:cannotimportname”libcafeconverter 这可以通过在Python3.6中使用虚拟环境来避免

解决方法:

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

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

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

现在它应该按计划工作了

相关问题 更多 >