Mercurial > ift6266
diff transformations/add_background_image.py @ 70:be24db3a4d6e
bug fix in reading pkl image list file
author | Xavier Glorot <glorotxa@iro.umontreal.ca> |
---|---|
date | Wed, 10 Feb 2010 00:38:54 -0500 |
parents | 6d87c0df2b0e |
children | 53ee1097c02c |
line wrap: on
line diff
--- a/transformations/add_background_image.py Tue Feb 09 22:17:37 2010 -0500 +++ b/transformations/add_background_image.py Wed Feb 10 00:38:54 2010 -0500 @@ -16,8 +16,8 @@ self.h = 32 self.w = 32 self.threshold = threshold; - self.bg_image_file = '/data/lisa/data/ift6266h10/image_net/filelist.pkl' - f=open(self.bg_image_file) + self.bg_image_file = '/data/lisa/data/ift6266h10/image_net/' + f=open(self.bg_image_file+'filelist.pkl') self.image_files = cPickle.load(f) f.close() @@ -52,11 +52,11 @@ x, y = random.randint(0, i_w - self.w), random.randint(0, i_h - self.h) return image[x:x + self.w, y:y + self.h] - # select a random background image from "bg_image_dir" and crops it + # select a random background image from "bg_image_file" and crops it def rand_bg_image(self): i = random.randint(0, len(self.image_files) - 1) - image = self.load_image(self.bg_image_dir + self.image_files[i]) + image = self.load_image(self.bg_image_file + self.image_files[i]) self.bg_image = self.rand_crop(image) # set "bg_image" as background to "image", based on a pixels threshold