tensorflow-lite - CoreMl 是否支持输出长特征而不是最终预测？

Question

我正在制作一个需要模型来输出给定图像的长特征（如 1,2024）的应用程序。我用的是iphone，所以我选择了coreml。但是，如果有人知道哪个框架可以输出长特征而不是预测标签，那么也欢迎使用其他框架。

Answer 1

是的，Core ML 可以做到这一点。只是不要将其转换为“分类器”。Core ML 对分类器进行特殊处理，并为这些模型输出一个概率字典。但是，如果您的模型不是分类器，那么您会得到一个以特征向量作为输出的 MLMultiArray。