首页 > 解决方案 > 如何判断图像是不是东西?

问题描述

我有一个训练有素的模型,可以确定图像是否包含猫或狗。我正在使用 SVM 进行分类。我想要一个新模型来确定图像是否是鼠标。它与分类为 2 类的第一个模型不同。此新模型​​将仅返回 TRUE 或 FALSE。

我认为我不能使用分类器模型,因为我只有 1 个类;鼠标。我不打算在第二个模型中使用第一个模型或与之相关的任何东西。

解决这个问题的最佳方法是什么?

标签: python-3.xmachine-learningcomputer-vision

解决方案


如果您希望能够预测图像是否有鼠标,这仍然是一个分类问题,您将图像分为两类 - 鼠标和非鼠标。

您打算使用的任何分类器都可能需要学习一个边界,以在 n 维输入空间中将鼠标图像与非鼠标图像分开。

如果您使用神经网络,您可以只使用一个输出神经元,理想情况下,当图像是鼠标时,它会给出高概率,而当图像中没有鼠标时,它会给出低概率。


推荐阅读