python - logistic - ¿Cómo crear un Dataset de imágenes como el conjunto de datos MNIST?

logistic regression example

Question

Tengo 10000 imágenes BMP de algunos dígitos escritos a mano. Si quiero alimentar los datos a una red neuronal, ¿qué debo hacer? Para el conjunto de datos de MNIST, solo tenía que escribir

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

Estoy usando la biblioteca de Keras en Python. ¿Cómo puedo crear ese conjunto de datos?

Answer 1

Puede escribir una función que cargue todas sus imágenes y apilarlas en una matriz numpy si todo cabe en la RAM o usar Keras ImageDataGenerator ( https://keras.io/preprocessing/image/ ) que incluye una función flow_from_directory . Puede encontrar un ejemplo aquí https://gist.github.com/fchollet/0830affa1f7f19fd47b06d4cf89ed44d .