Wie erstellt man ein Bild-Dataset genau wie das MNIST-Dataset?

Question

Ich habe 10000 BMP Bilder von einigen handschriftlichen Ziffern. Wenn ich die Daten einem neuronalen Netzwerk zuführen möchte, was muss ich tun? Für MNIST-Dataset musste ich einfach schreiben

(X_train, y_train), (X_test, y_test) = mnist.load_data() 

Ich benutze Keras-Bibliothek in Python. Wie kann ich einen solchen Datensatz erstellen?

Answer 1

Sie können entweder eine Funktion schreiben, die alle Ihre Bilder laden und stapeln sie in eine numpy Array, wenn alle im RAM passen oder verwenden Keras ImageDataGenerator (https://keras.io/preprocessing/image/), die flow_from_directory eine Funktion enthält. Hier finden Sie ein Beispiel https://gist.github.com/fchollet/0830affa1f7f19fd47b06d4cf89ed44d.