在python中读取(SVHN)数据集



我想使用python读取以下数据集:http://ufldl.stanford.edu/housenumbers/train_32x32.mat

我使用io.loadmat('rain_32x32.mat')加载了这个mat文件,但当我试图显示上面numpy数组中的图像时,我无法获得具有良好分辨率和着色的图像。

知道如何从这个数据集中读取和绘制图像吗?

loadmat函数的输出是一个字典。请参阅以下代码:
import numpy as np
import scipy.io as sio
import matplotlib.pyplot as plt
%matplotlib inline
image_ind = 10
train_data = sio.loadmat('train_32x32.mat')
# access to the dict
x_train = train_data['X']
y_train = train_data['y']
# show sample
plt.imshow(x_train[:,:,:,image_ind])
plt.show()

print y_train[image_ind]

最新更新