A Demo of Convert Caffe Model into CoreML Model

阿娇IOS

于 2019-05-19 04:40:13 发布

阅读量206

点赞数

CC 4.0 BY-SA版权

分类专栏： Machine Learning

本文链接：https://blog.youkuaiyun.com/weixin_40515816/article/details/90338795

本文演示如何将Caffe深度学习模型转换为Apple的CoreML格式，以便在iOS和macOS应用中无缝使用。通过转换过程，你可以利用CoreML的便利性和设备上的推理能力，提高应用程序的效率。

import coremltools

#oxford102.caffemodel是模型名字
caffe_model = ('oxford102.caffemodel','deploy.prototxt')

#标签文件名
labels = 'flower-labels.txt'

coreml_model = coremltools.converters.caffe.convert (
    caffe_model,
    class_labels = labels,