python从date目录导入数据集_如何从特定目录或文件夹导入预下载的MNIST数据集？...

我将演示如何从头开始加载它（以便更好地理解），并演示如何通过matplotlib.pyplot从中显示数字图像import cPickle

import gzip

import numpy as np

import matplotlib.pyplot as plt

def load_data():

path = '../../data/mnist.pkl.gz'

f = gzip.open(path, 'rb')

training_data, validation_data, test_data = cPickle.load(f)

f.close()

X_train, y_train = training_data[0], training_data[1]

print X_train.shape, y_train.shape

# (50000L, 784L) (50000L,)

# get the first image and it's label

img1_arr, img1_label = X_train[0], y_train[0]

print img1_arr.shape, img1_label

# (784L,) , 5

# reshape first image(1 D vector) to 2D dimension image

img1_2d = np.reshape(img1_arr, (28, 28))

# show it

plt.subplot(111)

plt.imshow(img1_2d, cmap=plt.get_cmap('gray'))

plt.show()

您还可以通过以下示例函数将标签矢量化为a 10-dimensional unit vector：def vectorized_result(label):

e = np.zeros((10, 1))

e[label] = 1.0

return e

将上述标签矢量化：print vectorized_result(img1_label)

# output as below:

[[ 0.]

[ 0.]

[ 0.]

[ 0.]

[ 0.]

[ 1.]

[ 0.]

[ 0.]

[ 0.]

[ 0.]]

如果要将其转换为CNN输入，可以按如下方式进行重塑：def load_data_v2():

path = '../../data/mnist.pkl.gz'

f = gzip.open(path, 'rb')

training_data, validation_data, test_data = cPickle.load(f)

f.close()

X_train, y_train = training_data[0], training_data[1]

print X_train.shape, y_train.shape

# (50000L, 784L) (50000L,)

X_train = np.array([np.reshape(item, (28, 28)) for item in X_train])

y_train = np.array([vectorized_result(item) for item in y_train])

print X_train.shape, y_train.shape

# (50000L, 28L, 28L) (50000L, 10L, 1L)