CoreML 图像模型只识别给定的数据集，如何包含外部数据？

Posted 2023-03-28

【中文标题】CoreML 图像模型只识别给定的数据集，如何包含外部数据？

【英文标题】：CoreML image model only recognizing the given data sets, how to include outside data?

【发布时间】：2021-03-22 23:46:24

【问题描述】：

我有一个训练有素的 CoreML 图像模型，应该可以识别支票的正面和背面。我面临一个问题，模型将看到长颈鹿的图像，而模型只是试图确定图像是支票的正面还是支票的背面（置信度总计 100%）。理想情况下，模型不会说“这是支票背面与支票正面的可能性有多大”，而是“这是支票背面吗”。

【参考方案1】：

如果您的模型仅接受支票图像的训练，则您只能为其提供支票图像。如果你给它一些其他图像，它会认为这是一张支票，因为这是它唯一知道的事情。

要制作一个也可以检测“未检查”的模型，您需要添加一个新类别，并使用各种未检查对象的图像来训练模型。

或者，您可以使用某种 OOD（域外）检测来验证输入图像是否与模型训练过的事物相似。但这不是 Core ML 可以轻松做到的。