Swift：CoreML 中的图像预处理：调整大小和转置

Question

我需要在输入模型之前对 UIImage 对象进行预处理。

该模型需要 2 次转换：

1. 调整大小为 224,224
2. 转置为 CHW

我搜索并发现有两种方法：

1. 在 Swift 中进行预处理
2. 使用 Coremltools 将操作添加到 AI 模型

我只设法找到如何在 Swift 中调整大小，但是有不知道如何进行转置以确保图像具有 CHW 数据格式。

也不知道如何从 UIImage 中查看数据格式（CHW 或 HWC）。

如何转置以及如何查看数据格式？

Answer 1

设置图像输入类型，不用担心大小和数据格式。有明确的API。

例子：

image_input = ct.ImageType(name="input_image", shape=(224,224))
model = ct.convert(traced_model, inputs=[image_input])