準備Python環境
蘋果官網關于機器學習的介紹:https://developer.apple.com/cn/machine-learning/
在頁面中部找到模型轉換器
image.png
單擊進入python世界
image.png
在終端敲python --version
命令,蘋果要求的版本是2.7.13,而實際上2.7.14版本也可以正常運行
建議下載AnacondaPython 2.7版本
image.png
下載安裝好Anaconda后輸入以下命令創建一個名為flowerrec,Python版本為 2.7.13的空間
conda create --name flowerrec python=2.7.13
激活flowerrec空間
source activate flowerrec
接下來環境的前綴就會顯示(flowerrec)
字樣。
使用下面命令安裝CoreML工具(如果不識別pip命令,則先用easy_install pip
命令安裝pip)
pip install -U coremltools
生成mlmodel文件
- cd到包含oxford102.caffemodel、deploy.prototxt、class_labels.txt文件的文件夾,我的是
cd /Users/nick/Desktop/coffee\ model
。
輸入python
命令,然后在>>>
后輸入import coremltools
命令導入工具,如果出現更新TensorFlow版本提示無需理會。
輸入以下命令:
coreml_model = coremltools.converters.caffe.convert(('oxford102.caffemodel', 'deploy.prototxt'), image_input_names='data', class_labels='class_labels.txt')
其中 deploy.prototxt 文件是描述神經網絡的結構,oxford102.caffemodel 文件是coffe格式的數據訓練模型,class_labels.txt 文件是識別出來的花的種類,image_input_names='data' 表示輸入的數據,class_labels='class_labels.txt' 表示輸出結果。
- 出現
>>>
時,輸入以下命令來保存模型
coreml_model.save('Flowers.mlmodel')
- 當再次出現
>>>
時,表示模型已經生成完畢,原文件夾中多出一個mlmodel文件,接下來就可以用這個mlmodel文件了!!!
image.png