Ich versuche, Bilder in 10 Klassen zu klassifizieren. Um Wahrscheinlichkeiten für Bilder zu erhalten, verwende ich die Funktion model.predict_generator () in keras. Dies liefert nur Vorhersagewerte und nicht die entsprechende Proben-ID (in diesem Fall Bilddateiname).
%Vor%In welcher Reihenfolge liest das '.flow_from_directory' die Beispiele?
(ODER)
Wie finde ich den entsprechenden Bildnamen / die ID der Vorhersagen?
Klicken Sie hier für den Code .
Wenn Sie shuffle=False einstellen, sieht es so aus, als ob Bilder in der Reihenfolge zurückgegeben werden, in der os.listdir sie zurückgibt.
Ich habe das gerade getestet, indem ich mit meinem Generator ein paar Beispielbilder in eine Datei mit shuffle=False
Sie können auch die Dateinamen des Generators ausdrucken .. die (ich schaute auf den Code) führt nur os.listdir
print test_generator.filenames
Also sollte Ihr predict_generator Ergebnis bei Index 0 für das Bild bei os.listdir index 0 sein.
Tags und Links python-2.7 keras