Mercurial > ift6266
comparison scripts/creer_jeu_occlusion.py @ 153:041d68b34b70
Script utilise pour creer le jeu de donnees utilise pour le module Occlusion.py. De plus, le fichier /data/lisa/data/ift6266h10/echantillon_occlusion.py a change afin d'etre cree avec seed random (jeu reproductible)
| author | SylvainPL <sylvain.pannetier.lebeuf@umontreal.ca> |
|---|---|
| date | Wed, 24 Feb 2010 13:51:18 -0500 |
| parents | |
| children |
comparison
equal
deleted
inserted
replaced
| 152:8ab3ae1b0148 | 153:041d68b34b70 |
|---|---|
| 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() |