diff transformations/add_background_image.py @ 68:ee6a788557b6

Change in add_background_image to load a pickled list of file, not to do a listdir
author Xavier Glorot <glorotxa@iro.umontreal.ca>
date Tue, 09 Feb 2010 22:16:22 -0500
parents ab70fbca513c
children 6d87c0df2b0e
line wrap: on
line diff
--- a/transformations/add_background_image.py	Tue Feb 09 21:33:57 2010 -0500
+++ b/transformations/add_background_image.py	Tue Feb 09 22:16:22 2010 -0500
@@ -7,7 +7,8 @@
     Author: Guillaume Sicard
 '''
 
-import sys, os, random, fnmatch
+import sys, os, random
+import cPickle
 import Image, numpy           
 
 class AddBackground():
@@ -15,9 +16,9 @@
         self.h = 32
         self.w = 32
         self.threshold = threshold;
-        self.bg_image_dir = '/data/lisa/data/ift6266h10/image_net/'
-        self.pattern = '*.JPEG'
-        self.image_files = fnmatch.filter(os.listdir(self.bg_image_dir),self.pattern)
+        self.bg_image_file = '/data/lisa/data/ift6266h10/image_net/filelist.pkl'
+        f=open(self.bg_image_file)
+        self.image_files = cPickle.load(f)
 
     # get threshold value
     def get_settings_names(self):