-1

我想使用 CreateML 创建一个图像分类器模型。我有非常高分辨率的图像,但在数据流量和处理时间方面是有代价的,所以我更喜欢使用尽可能小的图像。

文档说:

图像 (...) 不必是特定大小,也不必彼此大小相同。但是,最好使用至少299 x 299 像素的图像。

我用各种尺寸 > 299x299px 的图像训练了一个测试模型,Xcode 中的模型参数显示尺寸为 299x299px,据我所知,这是标准化的图像尺寸:

在此处输入图像描述

这个维度似乎是由 CreateML Image Classifier 算法决定的,不可配置。

  1. 用大于 299x299px 的图像训练模型是否有意义?
  2. 如果图像尺寸不是正方形(与宽度相同的高度),在归一化过程中训练图像是否会被中心裁剪为 299x299px,或者图像在正方形之外的部分是否会影响模型?
4

1 回答 1

1

根据阅读和训练图像分类模型的经验(但没有直接了解 Apple 内部知识),似乎 Create ML 会缩放传入的图像以适合 299 x 299 的方形图像。提供更大的图像会浪费磁盘空间和预处理时间。我能找到的最佳文档是查看 CreateML 为图像分类器模板创建的 mlmodel 文件。输入被明确定义为彩色图像 299 x 299。在独立应用程序中没有更改该设置的选项。这是一些文档(适用于默认使用 ScenePrint 的 Classifer 模板):

https://developer.apple.com/documentation/createml/mlimageclassifier/featureextractortype/sceneprint_revision

Playground 工作区中可能有一个 Center/Crop 选项,但我从未在 Create ML 的独立应用程序版本中找到它。

于 2021-01-26T12:43:52.033 回答