Mercurial > ift6266

comparison scripts/creer_jeu_occlusion.py @ 159:e81241cfc2de

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
merge
author Myriam Cote <cotemyri@iro.umontreal.ca>
date Thu, 25 Feb 2010 09:05:48 -0500
parents 041d68b34b70
children
comparison
equal deleted inserted replaced
158:d1bb6e06497a 159:e81241cfc2de
1 #!/usr/bin/python
2 # coding: utf-8
3
4 '''
5 Sert a creer un petit jeu de donnees afin de pouvoir avoir des fragments
6 de lettres pour ajouter bruit d'occlusion
7
8 Sylvain Pannetier Lebeuf dans le cadre de IFT6266, hiver 2010
9
10 '''
11
12 from pylearn.io import filetensor as ft
13 import pylab
14 import random as r
15 from numpy import *
16
17 nombre = 20000 #La grandeur de l'echantillon
18
19 f = open('all_train_data.ft') #Le jeu de donnees est en local.
20 d = ft.read(f)
21 f.close()
22 print len(d)
23 random.seed(3525)
24
25 echantillon=r.sample(xrange(len(d)),nombre)
26 nouveau=d[0:nombre]
27 for i in xrange(nombre):
28 nouveau[i]=d[echantillon[i]]
29
30
31 f2 = open('echantillon_occlusion.ft', 'w')
32 ft.write(f2,nouveau)
33 f2.close()
34
35
36 ##Tester si ca a fonctionne
37 f3 = open('echantillon_occlusion.ft')
38
39 d2=ft.read(f3)
40 pylab.imshow(d2[0].reshape((32,32)))
41 pylab.show()
42 f3.close()