使用 CreateML 应用创建 CoreML 模型的指南 — 图像分类
![](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/c85559ec572bdaa8e43e82a6832e34f0.png)
CoreML 框架使得将机器学习模型集成到 Apple 平台应用中变得更加容易。
虽然有多种方法可以创建 CoreML 模型,但使用 Xcode 中的 CreateML 开发者工具无疑是最简单的方法之一。
我们使用一个著名的猫狗数据集,来展示如何使用 CreateML 创建图像分类模型。
数据集下载地址:Microsoft 下载链接
1. 启动 CreateML 应用
首先,打开 Xcode。然后在菜单栏中点击 Xcode,选择“打开开发者工具”,然后点击“CreateML”。此时会出现一个查找器窗口,选择新文档以启动一个新的 CreateML 项目。
![启动 CreateML](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/bacba63e72848128612b125bc39bbc40.gif)
2. 选择模型类型
在模板选择页面中,选择图像分类模型类型,然后点击“下一步”。
![选择模型类型](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/63acf3cae751d680127a680dd18cc256.png)
3. 指定模型属性
输入模型的名称和作者,然后选择一个合适的文件夹来保存 CreateML 项目。
![指定模型属性](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/4afb49b8303b65c39ba9f97ebbff0508.png)
4. 设置页面
在训练机器学习模型时,设置页面可能看起来有些复杂,但实际上非常简单。
在数据部分,有三个部分:训练数据、验证数据和测试数据。训练数据用于训练模型,验证数据用于评估模型在训练过程中的表现,测试数据用于最终评估模型的准确性。
在参数部分,主要设置最大迭代次数,即训练过程中数据通过模型的最大次数。
在增强部分,可以在数据传递给模型之前进行一些编辑,以防止模型过拟合。
![设置页面](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/3b0ce92b31cbb276d8262fdafb552a96.png)
5. 添加训练数据
将猫狗数据集按照要求的布局组织好,拖到训练数据部分。由于数据量大,加载时间可能较长。
![添加训练数据](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/fefe3a2785274b86481c46737cd3b3f4.png)
6. 开始训练
无需数据扩充,点击窗口左上角的播放按钮开始训练。训练时间因 Mac 的性能而异。
![开始训练](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/3398841f7bed5d339484a70eefa2dac2.png)
![训练过程](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/4d2c0d2723b25d96f03e7689be241b44.png)
7. 评估和准确性
训练完成后,评估页面显示每个类别的准确率和召回率。这些指标用于评估模型的性能。
![评估和准确性](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/231c86d6075eb30e0fd977cf5f47c685.png)
预览页面允许您使用自己的图像测试模型,查看预测结果。
![预览页面-cat](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/38aba34510faf0f656db5da967c3c220.png)
![预览页面-dog](https://raw.githubusercontent.com/lijianfeigeek/PictureBed/main/uPic/f4075a4b7e53f98cc259d78295dbb043.png)
8. 导出模型
训练完成后,可以导出模型以便在应用中使用。点击输出页面中的“获取”按钮,或者直接发送到 Xcode 项目。