python - 在 python 中读取 (SVHN) 数据集

Question

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

我已经使用 io.loadmat('train_32x32.mat') 加载了这个 mat 文件，但是当我尝试显示来自上述 numpy 数组的图像时，我没有得到具有良好分辨率和着色的图像。

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

Answer 1

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]