Small script for extracting images out of the svhn dataset

svhn.py

import scipy.io
import numpy as np
from PIL import Image

np.set_printoptions(threshold=np.nan)

mat = scipy.io.loadmat('train_32x32.mat')
arr =  np.array(mat["X"])

for i in range(5000):
	img =  Image.fromarray(arr[:,:,:,i], 'RGB')
	img.save("svhn"+str(i)+".jpg")
	print "saving: " + str(i)