You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何导出支持任意尺寸输入的YOLO分割模型为.mlpackage/.mlmodel文件用于Xcode

如何导出支持任意尺寸输入的YOLO分割模型为.mlpackage/.mlmodel文件用于Xcode

别担心,我一步步教你修改模型,让它支持任意尺寸的图像输入,完美适配Xcode的需求:

步骤1:导出基础的CoreML模型

先按你原来的代码导出初始的CoreML模型,这一步会得到固定640x640输入尺寸的.mlpackage文件:

from ultralytics import YOLO

# 加载YOLO11n分割预训练模型
model = YOLO("yolo11n-seg.pt")

# 导出为CoreML格式,生成初始模型文件
model.export(format="coreml")

步骤2:用coremltools修改输入尺寸为无界范围

接下来我们用coremltools调整模型的输入规格,让它支持任意大小的图像:

import coremltools as ct

# 加载刚导出的初始CoreML模型(记得替换成你实际生成的模型文件名)
coreml_model = ct.models.MLModel("yolo11n-seg.mlpackage")

# 获取模型的输入描述信息
input_spec = coreml_model.get_spec().description.input[0]

# 设置输入图像的宽度和高度为无界范围:最小1像素,最大无限制
input_spec.type.imageType.widthRange.min = 1
input_spec.type.imageType.widthRange.max = ct.UNBOUNDED
input_spec.type.imageType.heightRange.min = 1
input_spec.type.imageType.heightRange.max = ct.UNBOUNDED

# 保存修改后的模型,文件名可以自己命名,比如加个-unbounded后缀
coreml_model.save("yolo11n-seg-unbounded.mlpackage")

补充说明

  • 这样修改后,模型就能接受任意尺寸的图像输入了,导入Xcode后无需再限制图像大小
  • YOLO模型内部会自动处理不同尺寸输入的预处理逻辑,你不用额外担心缩放问题
  • 如果你需要导出.mlmodel格式而不是.mlpackage,可以在model.export()里加上bundle=False参数,后续修改步骤完全一致

备注:内容来源于stack exchange,提问作者J. Doe

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.04.14 17:03:05