Как создать набор данных изображений так же, как набор данных MNIST?

У меня есть 10000 BMP изображений некоторых рукописных цифр. Если я хочу передать данные в нейронную сеть, что мне нужно сделать ? Для набора данных MNIST мне просто нужно было написать

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

Я использую библиотеку Keras в python . Как создать такой набор данных ?

1 ответов


вы можете написать функцию, которая загружает все ваши изображения и складывает их в массив numpy, если все помещается в ОЗУ, или использовать Keras ImageDataGenerator (https://keras.io/preprocessing/image/), который включает в себя функцию flow_from_directory. Вы можете найти пример здесь https://gist.github.com/fchollet/0830affa1f7f19fd47b06d4cf89ed44d.