comparison transformations/pipeline.py @ 52:c89defea1e65

Modification aux déformations élastiques pour mettre en cache 50x10 champs de déformation, selon 10 niveaux de complexité
author fsavard
date Thu, 04 Feb 2010 16:53:50 -0500
parents ff59670cd1f9
children cc4be6b25b8e
comparison
equal deleted inserted replaced
51:81b9567ec4ae 52:c89defea1e65
12 import random 12 import random
13 13
14 # To debug locally, also call with -s 1 (to stop after 1 batch ~= 100) 14 # To debug locally, also call with -s 1 (to stop after 1 batch ~= 100)
15 # (otherwise we allocate all needed memory, might be loonnng and/or crash 15 # (otherwise we allocate all needed memory, might be loonnng and/or crash
16 # if, lucky like me, you have an age-old laptop creaking from everywhere) 16 # if, lucky like me, you have an age-old laptop creaking from everywhere)
17 DEBUG = False 17 DEBUG = True
18 DEBUG_X = False 18 DEBUG_X = False
19 if DEBUG: 19 if DEBUG:
20 DEBUG_X = False # Debug under X (pylab.show()) 20 DEBUG_X = False # Debug under X (pylab.show())
21 21
22 DEBUG_IMAGES_PATH = None 22 DEBUG_IMAGES_PATH = None
55 # Either put the visualizer as in the MODULES_INSTANCES list 55 # Either put the visualizer as in the MODULES_INSTANCES list
56 # after each module you want to visualize, or in the 56 # after each module you want to visualize, or in the
57 # AFTER_EACH_MODULE_HOOK list (but not both, it's redundant) 57 # AFTER_EACH_MODULE_HOOK list (but not both, it's redundant)
58 VISUALIZER = Visualizer(to_dir=DEBUG_OUTPUT_DIR, on_screen=False) 58 VISUALIZER = Visualizer(to_dir=DEBUG_OUTPUT_DIR, on_screen=False)
59 59
60 MODULE_INSTANCES = [Thick(), LocalElasticDistorter(), PoivreSel(), Contrast()] 60 MODULE_INSTANCES = [LocalElasticDistorter()]
61 61
62 # These should have a "after_transform_callback(self, image)" method 62 # These should have a "after_transform_callback(self, image)" method
63 # (called after each call to transform_image in a module) 63 # (called after each call to transform_image in a module)
64 AFTER_EACH_MODULE_HOOK = [] 64 AFTER_EACH_MODULE_HOOK = []
65 if DEBUG: 65 if DEBUG: