将 .caffemodel 转换为 .mlmodel 失败

Posted

技术标签:

【中文标题】将 .caffemodel 转换为 .mlmodel 失败【英文标题】:Failed convert .caffemodel to .mlmodel 【发布时间】:2021-07-09 01:56:26 【问题描述】:

在尝试将 caffemodel 转换为 mlmodel 时,我无法运行我的 converter-script.py 这是我的 converter-script.py 文件:

import coremltools
caffe_model = ('oxford102.caffemodel', 'deploy.prototxt')
labels = 'flower-labels.txt'
models = coremltools.converters.caffe.converts(
   caffe_model,
   class_labels = labels,
   image_input_names = 'data'
)
coreml_model.save('FlowerClassifier.mlmodel')

我使用带有 python 2.7 的 virtualenv 运行它 我收到此错误消息:

文件“convert-script.py”,第 1 行,在 导入 coremltools 文件“/Users/aji/Documents/Environments/python27/lib/python2.7/site-packages/coremltools/init.py”,第 28 行,在 _root_logger_handlers_backup = _root_logger.handlers.copy() AttributeError: 'list' 对象没有属性 'copy'

谁能给我解决办法?

【问题讨论】:

【参考方案1】:

使用 python3 而不是从 python27 venv 创建和运行。

python3 convert-script.py

为我工作

【讨论】:

以上是关于将 .caffemodel 转换为 .mlmodel 失败的主要内容,如果未能解决你的问题,请参考以下文章

如何加载 caffe 模型并转换为 numpy 数组?

使用 iOS 11 mlmodel 进行图像分类 - 使用 coremltools 和经过训练的 .caffemodel 转换问题

Yolov8从pytorch到caffe 训练模型并转换到caffemodel

Caffe 到 CoreML 模型的转换

caffe中的caffemodel参数提取方法

caffemodel的读取与修改