使用Keras on Tensorflow进行图像分类:如何在训练期间找到哪些图像被错误分类?

时间:2017-06-24 20:01:19

标签: tensorflow deep-learning keras tensorboard ubuntu-17.04

我在Ubuntu 17.04上使用Keras 2.0(TensorFlow后端)进行二进制图像分类。一切都很好,除了我想看哪些图像被错误分类。我该怎么做?

另外,不确定它是否能解答我的问题,但是在TensorBoard中我无法使用image标签,所以不知道是否有帮助

我当然做了很多谷歌搜索,但我找不到答案。

1 个答案:

答案 0 :(得分:4)

只需预测分类并与您的真实价值进行比较......

predicted = model.predict(trainingImages)    

减去和删除符号应该导致正确的结果接近于零,而错误结果的结果很高:

result = numpy.absolute(trainingClasses-predicted)