view scripts/nist_read2.py @ 118:0d083964af4b

small change to testtransformation to match with the pipeline behavior
author Xavier Glorot <glorotxa@iro.umontreal.ca>
date Wed, 17 Feb 2010 16:25:44 -0500
parents a9b87b68101d
children
line wrap: on
line source

#!/usr/bin/env python

from pylearn.io import filetensor as ft
import pylab, numpy

datapath = '/data/lisa/data/ift6266h10/train_'

f = open(datapath+'data.ft')
d = ft.read(f)

f = open(datapath+'labels.ft')
labels = ft.read(f)

def label2chr(l):
    if l<10:
        return chr(l + ord('0'))
    elif l<36:
        return chr(l-10 + ord('A'))
    else:
        return chr(l-36 + ord('a'))

for i in range(min(d.shape[0],30)):
    pylab.figure()
    pylab.title(label2chr(labels[i]))
    pylab.imshow(d[i].reshape((32,32))/255., pylab.matplotlib.cm.Greys_r, interpolation='nearest')

pylab.show()