Wie erzeuge ich Tensorflow Confusion Matrix aus "output_graph.pb" und "output_labels.txt"?

Question

Also habe ich dieses Tutorial verfolgt und meine eigenen Bilder umgeschult.

Also habe ich jetzt eine "output_graph.pb" und eine "output_labels.txt" (die ich mit anderem Code verwenden kann, um Bilder zu klassifizieren).

Aber wie erzeuge ich tatsächlich eine Konfusionsmatrix mit einem Ordner zum Testen von Bildern (oder zumindest mit den Bildern, auf denen sie trainiert wurden)?

Es gibt https://www.tensorflow.org/api_docs/python/tf/confusion_matrix aber das scheint nicht sehr hilfreich.

Dieser Thread scheint nur zu sein, Zahlen mit Etikett darzustellen, anstatt tatsächliche Dateien, aber nicht wirklich sicher: how to create confusion matrix for classification in tensorflow

Und Im nicht wirklich sicher, wie Sie den Code in diesem Thread verwenden entweder: How do i create Confusion matrix of predicted and ground truth labels with Tensorflow?

Answer 1

ich würde versuchen Sie Konfusionsmatrix manuell zu erstellen, so etwas wie dies Schritten:

• das label_image Beispiel ändern nur das obere Etikett zu drucken.
• Schreiben Sie ein Skript, um das geänderte label_image für alle Bilder in einem Ordner wiederholt aufzurufen.
• Lassen Sie das Skript die Grundwahrheitsbeschriftung ausdrucken, und rufen Sie dann label_image auf, um die vorhergesagte zu drucken.

Sie sollten nun eine Textliste aller Ihrer Etiketten in der Konsole, so etwas wie dieses:

Apfel, Apfel Apfel, Birne Birne, Birne Birne, Orange ...

nun eine Tabelle mit Zeilen- und Spaltennamen für alle Etiketten erstellen:

 | apple | pear | orange 
-------+---------------------- 
apple | 
pear | 
orange | 

der Wert für jede Zelle wil l Die Anzahl der Paare, die in Ihrer Konsolenliste für Zeile, Spalte angezeigt werden. Für eine kleine Menge von Bildern können Sie dies manuell berechnen, oder Sie können ein Skript schreiben, um dies zu berechnen, wenn es zu viele gibt.