我正在使用迁移学习解决图像分类问题。现在,我在训练数据和测试数据上的准确率分别为 75% 和 67%。现在我想了解图像的哪些部分/部分用于以图形方式进行分类。
是否有任何包可以返回其中一个输入图像的副本,并在图像的最有用的像素或部分上带有标记?
我正在使用迁移学习解决图像分类问题。现在,我在训练数据和测试数据上的准确率分别为 75% 和 67%。现在我想了解图像的哪些部分/部分用于以图形方式进行分类。
是否有任何包可以返回其中一个输入图像的副本,并在图像的最有用的像素或部分上带有标记?
是的,典型的方法是获取输入的显着性图,这是每个像素对最终分类的贡献的“热图”。
在这本关于 Explainable ML 的免费在线书籍中,您可以找到获取显着性图最相关的方法,例如普通梯度,以及其他像素归因技术。
在这里,您可以找到有关如何使用 Keras 实现显着图的端到端教程。