Mercurial > ift6266
view scripts/creer_jeu_occlusion.py @ 162:050c7ff6b449
Do not commit syntax errors. (in test.py)
author | Arnaud Bergeron <abergeron@gmail.com> |
---|---|
date | Thu, 25 Feb 2010 18:12:18 -0500 |
parents | 041d68b34b70 |
children |
line wrap: on
line source
#!/usr/bin/python # coding: utf-8 ''' Sert a creer un petit jeu de donnees afin de pouvoir avoir des fragments de lettres pour ajouter bruit d'occlusion Sylvain Pannetier Lebeuf dans le cadre de IFT6266, hiver 2010 ''' from pylearn.io import filetensor as ft import pylab import random as r from numpy import * nombre = 20000 #La grandeur de l'echantillon f = open('all_train_data.ft') #Le jeu de donnees est en local. d = ft.read(f) f.close() print len(d) random.seed(3525) echantillon=r.sample(xrange(len(d)),nombre) nouveau=d[0:nombre] for i in xrange(nombre): nouveau[i]=d[echantillon[i]] f2 = open('echantillon_occlusion.ft', 'w') ft.write(f2,nouveau) f2.close() ##Tester si ca a fonctionne f3 = open('echantillon_occlusion.ft') d2=ft.read(f3) pylab.imshow(d2[0].reshape((32,32))) pylab.show() f3.close()