diff --git a/lib/pairwise_transform_utils.lua b/lib/pairwise_transform_utils.lua index 684ba52..57508b1 100644 --- a/lib/pairwise_transform_utils.lua +++ b/lib/pairwise_transform_utils.lua @@ -56,8 +56,8 @@ function pairwise_transform_utils.preprocess(src, crop_size, options) dest = pairwise_transform_utils.crop_if_large(dest, math.max(crop_size * 2, options.max_size)) dest = data_augmentation.blur(dest, options.random_blur_rate, options.random_blur_size, - options.random_blur_min, - options.random_blur_max) + options.random_blur_sigma_min, + options.random_blur_sigma_max) dest = data_augmentation.color_noise(dest, options.random_color_noise_rate) dest = data_augmentation.overlay(dest, options.random_overlay_rate) dest = data_augmentation.unsharp_mask(dest, options.random_unsharp_mask_rate)