浏览代码

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.overlay(dest, options.random_overlay_rate)
       dest = data_augmentation.unsharp_mask(dest, options.random_unsharp_mask_rate)
+      dest = iproc.crop_mod4(dest)
    else
       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))