Core ML工具安装:
pip install -U coremltools
Caffe模型转换到Core ML模型
下载一个caffe模型:
# Age and gender models
wget http://www.openu.ac.il/home/hassner/projects/cnn_agegender/cnn_age_gender_models_and_data.0.0.2.zip
unzip -a cnn_age_gender_models_and_data.0.0.2.zip
进行转换:
import coremltools
folder = 'cnn_age_gender_models_and_data.0.0.2'
coreml_model = coremltools.converters.caffe.convert(
(folder + '/age_net.caffemodel', folder + '/deploy_age.prototxt'),
image_input_names = 'data',
class_labels = 'ages.txt'
)
coreml_model.author = 'Gil Levi and Tal Hassner'
coreml_model.license = 'Unknown'
coreml_model.short_description = 'Age Classification using Convolutional Neural Networks'
coreml_model.input_description['data'] = 'An image with a face.'
coreml_mod