如何判断一张图片是否是飞机或是否是铅笔?

Question

用户上传图片时需要判断这张图是否属于某类图片，比如飞机。
我找到的示例是一个图片分类器，比如训练三个分类飞机、汽车、摩托，然后上传一张图片判断这张图片最接近哪一类。但是我并不确定用户会上传什么上来，比如我自己测试上传了一张猫，而回显的答案是飞机。显然我不可能训练所有图片分类，这也不符合我们的需求。
希望能只训练飞机一个分类，然后上传图片时判断是或否。
示例中的代码是用了sckikit-learn的极限森林算法聚类,但我查了这个api,没有找到合用的方法.
Methods

Methods

fit(X[, y])    Compute k-means clustering.
fit_predict(X[, y])    Compute cluster centers and predict cluster index for each sample.
fit_transform(X[, y])    Compute clustering and transform X to cluster-distance space.
get_params([deep])    Get parameters for this estimator.
predict(X)    Predict the closest cluster each sample in X belongs to.
score(X[, y])    Opposite of the value of X on the K-means objective.
set_params(**params)    Set the parameters of this estimator.
transform(X)    Transform X to a cluster-distance space.

请问应该怎么解决呢?

Answer 1

假如你训练了飞机、汽车、摩托 三个分类，那么用户传图片，你predict(X)得到应该是onehot的3个概率，你可以手动设置一下概率低于0.9或者某值时未识别。