Mercurial > ift6266

comparison data_generation/amt/amt_generate.py @ 400:8973abe35a9d

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Adding comments and fixing out of bounds index
author humel
date Wed, 28 Apr 2010 00:49:41 -0400
parents ae8102aad586
children
comparison
equal deleted inserted replaced
399:99905d9bc9dd 400:8973abe35a9d
30 resulting_data = numpy.zeros((total,IMGSHP[0]*IMGSHP[1])) 30 resulting_data = numpy.zeros((total,IMGSHP[0]*IMGSHP[1]))
31 resulting_labels = numpy.zeros((total,)) 31 resulting_labels = numpy.zeros((total,))
32 f.close();g.close() 32 f.close();g.close()
33 33
34 ds_size = len(test_data) 34 ds_size = len(test_data)
35 rand_seq = numpy.random.random_integers(ds_size, size=(NUM_BATCHES,BATCH_SIZE)) 35 rand_seq = numpy.random.random_integers(ds_size-1, size=(NUM_BATCHES,BATCH_SIZE))
36 36
37 for i in range(NUM_BATCHES): 37 for i in range(NUM_BATCHES):
38 for j in range(BATCH_SIZE): 38 for j in range(BATCH_SIZE):
39 resulting_data[i*BATCH_SIZE+j]=test_data[rand_seq[i,j]] 39 resulting_data[i*BATCH_SIZE+j]=test_data[rand_seq[i,j]]
40 resulting_labels[i*BATCH_SIZE+j] = test_labels[rand_seq[i,j]] 40 resulting_labels[i*BATCH_SIZE+j] = test_labels[rand_seq[i,j]]
62 all_images = [] 62 all_images = []
63 63
64 white_space = numpy.asarray(numpy.zeros((IMGSHP[0],WHITE_SPACE_THICKNESS))+255.,dtype='uint8') 64 white_space = numpy.asarray(numpy.zeros((IMGSHP[0],WHITE_SPACE_THICKNESS))+255.,dtype='uint8')
65 65
66 for i in range(len(seq)): 66 for i in range(len(seq)):
67 all_images += [numpy.asarray(seq[i].reshape((IMGSHP))*255.,dtype='uint8')] 67 all_images += [numpy.asarray(seq[i].reshape((IMGSHP)),dtype='uint8')]
68 68
69 all_images_stacked = numpy.hstack(numpy.asarray([numpy.hstack((image,white_space)) for image in all_images])) 69 all_images_stacked = numpy.hstack(numpy.asarray([numpy.hstack((image,white_space)) for image in all_images]))
70 return Image.fromarray(all_images_stacked) 70 return Image.fromarray(all_images_stacked)
71 71
72 def generate_labels(seq, indexes): 72 def generate_labels(seq, indexes):