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

Question

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

Answer 1

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

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

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