如何加载自定义数据集供CNN使用？

Question

MNIST使用以下代码加载数据：

fashion_mnist = keras.datasets.fashion_mnist
(train_images, train_labels), (test_images, test_labels) = fashion_mnist.load_data()

我已经准备好了自定义数据集并想加载相同的数据集。所以，尝试了以下代码：

^{pr2}$

其中t10k-images-idx3-ubyte是自定义图像训练数据
t10k-labels-idx1-ubyte是自定义标签训练数据
test-images-idx3-ubyte是自定义图像测试数据
test-labels-idx1-ubyte是自定义标签测试数据

但它将其保存为(train_images, train_labels), (test_images, test_labels)中的字符串。 您能帮助我正确地将这些数据加载到
(train_images, train_labels), (test_images, test_labels)？在

Answer 1

在您的示例中，只需将字符串元组分配给变量： 在

train_images = 'C:\\Users\\sm50014\\Desktop\\new\\t10k-images-idx3-ubyte'
train_labels = 'C:\\Users\\sm50014\\Desktop\\new\\t10k-labels-idx1-ubyte'
test_images = 'C:\\Users\\sm50014\\Desktop\\new\\test-images-idx3-ubyte'
test_labels = 'C:\\Users\\sm50014\\Desktop\\new\\test-labels-idx1-ubyte'

要加载您自己的数据集，您应该创建自定义加载程序来读取图像并将其馈送到您的网络。您可以检查这样一个加载器的例子，例如：https://medium.com/@waleedka/traffic-sign-recognition-with-tensorflow-629dffc391a6（functionload_data）。在