MobileNet混淆矩阵

时间:2018-05-05 15:55:15

标签: neural-network

我在跟踪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可以给我混淆矩阵,否则我应该手动完成所有,因为脚本重新训练可以打印来自测试集的所有错误分类的图像,但它将是一个巨大的工作要手动完成。

0 个答案:

没有答案