VGG: Reihenfolge der Kategorien?

Question

Ich benutze VGG-16-Netzwerk auf ILSVRC mit Fast-RCNN trainiert.

Sie können eine 1000-dimensionale Matrix für die Wahrscheinlichkeit von jeder der 1.000 Klassen in ILSVRC erhalten.

Ich kann jedoch nicht herausfinden, welche Spalte in der Matrix welcher Klasse entspricht.

Zum Beispiel würde ich gerne wissen, ob final_prob [2] Mensch oder Auto entspricht.

Es gibt eine Reihe von Klassen für ILSVRC jedes Jahr (http://image-net.org/challenges/LSVRC/2014/browse-synsets)

aber die Matrix scheint nicht in dieser Reihenfolge zu sein.

Wo kann ich herausfinden, wie die Einträge in der Matrix den Klassen zugeordnet sind?

Answer 1

Wenn Sie eine der vorher gelernten Modelle in Caffe verwenden, führen data/ilsvrc12/get_ilsvrc_aux.sh und dann die Liste der Namen von Klassen in data/ilsvrc12/synset_words.txt sein wird. Dies ist in der gleichen Reihenfolge wie die 1000-dimensionale Ausgabe des Modells.