我在跟踪MarkDaoust的gitHub存储库时遇到了一个问题:https://github.com/googlecodelabs/tensorflow-for-poets-2
我所做的是创建一个对不同类进行分类的mobilenet模型,创建了retrained_graph.pb和retrained_labels.txt
问题在于我无法为所有类创建混淆矩阵。我试过使用了这个脚本evaluate.py,但它没用,甚至tf.confusion_matrix也没用,因为我不知道测试集/ training_test中的图像是哪一个(我只知道将从中挑选多少图像)每个类文件夹)。
我希望有一个函数或类似的东西,因为retrained_graph.pb可以给我混淆矩阵,否则我应该手动完成所有,因为脚本重新训练可以打印来自测试集的所有错误分类的图像,但它将是一个巨大的工作要手动完成。