如何像MNIST数据集一样创建图像数据集？

Question

我有10000个BMP图像的一些手写数字.如果我想将数据馈送到神经网络,我需要做什么？对于MNIST数据集,我只需要写

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

我在python中使用Keras库.我该如何创建这样的数据集？

Answer 1

您可以编写一个加载所有图像的函数,并将它们堆叠成一个numpy数组(如果所有图像都适合RAM)或使用包含函数的Keras ImageDataGenerator(https://keras.io/preprocessing/image/)flow_from_directory.你可以在这里找到一个例子https://gist.github.com/fchollet/0830affa1f7f19fd47b06d4cf89ed44d.