Mercurial > ift6266
diff transformations/local_elastic_distortions.py @ 29:b67d729ebfe3
Adapted to avoid saving parameters completely determined by complexity
author | fsavard <francois.savard@polymtl.ca> |
---|---|
date | Fri, 29 Jan 2010 17:16:04 -0500 |
parents | 010e826b41e8 |
children | a8ac3402eb45 |
line wrap: on
line diff
--- a/transformations/local_elastic_distortions.py Fri Jan 29 14:12:09 2010 -0500 +++ b/transformations/local_elastic_distortions.py Fri Jan 29 17:16:04 2010 -0500 @@ -106,7 +106,12 @@ idx = numpy.random.randint(0, len(self.precomputed_params)) self.current_params = self.precomputed_params[idx] - return self.current_params.alpha_sigma() + # don't return anything, to avoid storing deterministic parameters + return [] # self.current_params.alpha_sigma() + + def get_parameters_determined_by_complexity(self, complexity): + tmp_params = self._initialize_new_params(complexity) + return tmp_params.alpha_sigma() # adapted from http://blenderartists.org/forum/showthread.php?t=163361 def _gen_gaussian_kernel(self, sigma): @@ -147,23 +152,25 @@ return params.alpha * field - def _initialize_new_params(self): + def _initialize_new_params(self, complexity=None): + if not complexity: + complexity = self.current_complexity + params = ElasticDistortionParams(self.image_size) - cpx = self.current_complexity # pour faire progresser la complexité un peu plus vite # tout en gardant les extrêmes de 0.0 et 1.0 - cpx = cpx ** (1./3.) + complexity = complexity ** (1./3.) # the smaller the alpha, the closest the pixels are fetched # a max of 10 is reasonable - params.alpha = cpx * 10.0 + params.alpha = complexity * 10.0 # the bigger the sigma, the smoother is the distortion # max of 1 is "reasonable", but produces VERY noisy results # And the bigger the sigma, the bigger the blur kernel, and the # slower the field generation, btw. - params.sigma = 10.0 - (7.0 * cpx) + params.sigma = 10.0 - (7.0 * complexity) return params @@ -308,7 +315,8 @@ img = _load_image(imgpath) dist = LocalElasticDistorter((32,32)) print dist.regenerate_parameters(0.5) - img = dist.distort_image(img) + img = dist.transform_image(img) + print dist.get_parameters_determined_by_complexity(0.4) pylab.imshow(img) pylab.show() @@ -419,8 +427,9 @@ import os.path #_distorter_tests() #_benchmark() - #_specific_test() + _specific_test() #_complexity_tests() - _complexity_benchmark() + #_complexity_benchmark() +