ソースを参照

Fix a bug in preprocessing when filters=box only

nagadomi 9 年 前
コミット
eeb90dc39b
1 ファイル変更1 行追加0 行削除
  1. 1 0
      lib/pairwise_transform_utils.lua

+ 1 - 0
lib/pairwise_transform_utils.lua

@@ -49,6 +49,7 @@ function pairwise_transform_utils.preprocess(src, crop_size, options)
       dest = data_augmentation.color_noise(dest, options.random_color_noise_rate)
       dest = data_augmentation.color_noise(dest, options.random_color_noise_rate)
       dest = data_augmentation.overlay(dest, options.random_overlay_rate)
       dest = data_augmentation.overlay(dest, options.random_overlay_rate)
       dest = data_augmentation.unsharp_mask(dest, options.random_unsharp_mask_rate)
       dest = data_augmentation.unsharp_mask(dest, options.random_unsharp_mask_rate)
+      dest = iproc.crop_mod4(dest)
    else
    else
       dest = pairwise_transform_utils.random_half(dest, options.random_half_rate, options.downsampling_filters)
       dest = pairwise_transform_utils.random_half(dest, options.random_half_rate, options.downsampling_filters)
       dest = pairwise_transform_utils.crop_if_large(dest, math.max(crop_size * 2, options.max_size))
       dest = pairwise_transform_utils.crop_if_large(dest, math.max(crop_size * 2, options.max_size))