|
@@ -26,11 +26,11 @@ class Transformation():
|
|
|
n_components = self.param
|
|
|
data = reconstruction.fast_ica(img, n_components)
|
|
|
|
|
|
- if self.transformation == 'diff_filter':
|
|
|
+ if self.transformation == 'min_diff_filter':
|
|
|
w_size, h_size = list(map(int, self.param.split(',')))
|
|
|
# bilateral with window of size (`w_size`, `h_size`)
|
|
|
lab_img = transform.get_LAB_L(img)
|
|
|
- data = convolution.convolution2D(lab_img, kernels.bilateral_diff, (w_size, h_size))
|
|
|
+ data = convolution.convolution2D(lab_img, kernels.min_bilateral_diff, (w_size, h_size))
|
|
|
|
|
|
if self.transformation == 'static':
|
|
|
# static content, we keep input as it is
|
|
@@ -54,7 +54,7 @@ class Transformation():
|
|
|
n_components = self.param
|
|
|
path = os.path.join(path, 'N' + str(n_components))
|
|
|
|
|
|
- if self.transformation == 'diff_filter':
|
|
|
+ if self.transformation == 'min_diff_filter':
|
|
|
w_size, h_size = list(map(int, self.param.split(',')))
|
|
|
path = os.path.join(path, 'W_' + str(w_size)) + '_' + str(h_size)
|
|
|
|