如何在Swift中集成CoreML CoreML模型转换与调用指南

新闻中心 NEWS CENTER

您当前位置：首页 > 新闻中心 > 行业资讯

如何在Swift中集成CoreML CoreML模型转换与调用指南

2025-08-14

浏览次数：次

返回列表

在swift中集成coreml模型的关键步骤包括模型准备、转换为.mlmodel格式、导入xcode项目以及编写调用逻辑。1. 准备模型时需确保其来自支持的框架（如keras、tensorflow）并已完成训练；2. 使用coremltools库将模型转换为.mlmodel格式，注意定义输入输出结构；3. 将.mlmodel文件拖入xcode项目，生成对应的swift类并检查模型输入输出配置；4. 在swift代码中加载模型、准备符合要求的输入数据（如cvpixelbuffer或mlmultiarray）、调用prediction方法并处理结果；5. 若遇到问题，应检查输入输出匹配性、预处理一致性及模型性能优化。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

如何在Swift中集成CoreML CoreML模型转换与调用指南

在Swift中集成CoreML模型，关键在于把训练好的机器学习模型转换成CoreML格式，并正确地在代码中调用。整个过程包括模型准备、转换、导入Xcode以及编写调用逻辑这几个核心步骤。

准备你的机器学习模型

CoreML支持多种主流的机器学习框架导出的模型，比如TensorFlow、Keras、Scikit-learn等。但这些模型不能直接使用，必须先转换为

.mlmodel

格式。

如果你使用的是Python环境，可以借助
```
coremltools
```
库来完成转换。
确保你的原始模型已经训练完成并保存为标准格式（如
```
.h5
```
、
```
.pkl
```
或
```
.pb
```
）。
转换前需要明确输入输出的数据类型和维度，这对后续在Swift中的调用非常重要。

例如，在Python中转换一个Keras模型：

import coremltools as ct

# 加载Keras模型
keras_model = load_model('my_model.h5')

# 转换为CoreML模型
ml_model = ct.converters.keras.convert(keras_model, input_names='input', output_names='output')

# 保存为.mlmodel文件
ml_model.s*e('MyModel.mlmodel')

将.mlmodel文件导入Xcode项目

这一步非常直观，但也有一些需要注意的地方。

在Xcode中打开你的项目，将生成的
```
.mlmodel
```
文件拖入项目导航器中。
Xcode会自动解析该模型，并在项目中生成对应的Swift类。
检查模型详情页，确认输入输出结构是否符合预期，比如图像尺寸、数据类型等。

如果你看到类似“Image (width x height)”这样的字段，说明这个模型是处理图像的，你需要确保传入的图片满足其尺寸要求。

N世界

一分钟搭建会展元宇宙

138 查看详情 N世界

在Swift中调用CoreML模型进行预测

一旦模型导入成功，就可以通过Swift代码进行调用了。这里以图像分类为例。

加载模型：通常不需要手动初始化，Xcode生成的类提供了默认构造函数。
准备输入数据：比如将
```
UIImage
```
转换为模型接受的
```
CVPixelBuffer
```
或
```
MLMultiArray
```
。
执行预测：调用模型的
```
prediction()
```
方法。
处理结果：根据输出解析预测结果。

示例代码片段如下：

guard let model = try? MyModel(configuration: MLModelConfiguration()) else { return }

let input = try? MyModelInput(image: pixelBuffer) // 假设pixelBuffer已准备好
if let prediction = try? model.prediction(input: input) {
    print("预测结果：\(prediction.output)")
}