代码之家 › 专栏 › 技术社区 › Franva

Swift:CoreML中的图像预处理:调整大小和转换

coremltools coreml image-processing swift

Franva · 技术社区 · 4 年前

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

该模型需要2个转换:

我搜索发现有两种方法:

我只找到了如何在Swift中调整大小,但不知道如何进行转置以确保图像的数据格式为CHW。

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

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

0 回复 | 直到 4 年前

Dmytro Hrebeniuk 4 年前

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

示例:

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

推荐文章

sudoExclamationExclamation · iOS自动布局-高度不能小于ThanOrEqualTo

1 年前

Tom · SwiftUI.sheet导致背景白色闪烁

1 年前

Danil · 种子/填充核心数据的最佳实践?[关闭]

1 年前

Z S · SwiftUI:使用@FocusState。与视图初始化器绑定

1 年前

Robin · LazyVGrid项目预计不会击中测试区域

1 年前

Mahendra · Popover不是从SwiftUI中的项目列表中打开的[重复]

1 年前

Saurabh Saini · 结构下标

1 年前

Alex Smith · 移动到下一个视图控制器后如何显示警报?

1 年前

jnwjack · Swift:初始化不安全可变指针<CGImage?>不分配

1 年前

Moses Harding · 在SwiftUI的ForEach中以绑定和非绑定方式使用数组的单个成员,而不使用索引

1 年前