فهرست منبع

Fix border removing

nagadomi 9 سال پیش
والد
کامیت
99e6dd1a57
2فایلهای تغییر یافته به همراه8 افزوده شده و 14 حذف شده
  1. 8 10
      lib/pairwise_transform_scale.lua
  2. 0 4
      lib/pairwise_transform_utils.lua

+ 8 - 10
lib/pairwise_transform_scale.lua

@@ -28,23 +28,21 @@ function pairwise_transform.scale(src, scale, size, offset, n, options)
 	 x = small
 	 x = small
       end
       end
    end
    end
-
+   local scale_inner = scale
+   if options.x_upsampling then
+      scale_inner = 1
+   end
+   x = iproc.crop(x, unstable_region_offset, unstable_region_offset,
+		  x:size(3) - unstable_region_offset, x:size(2) - unstable_region_offset)
+   y = iproc.crop(y, unstable_region_offset * scale_inner, unstable_region_offset * scale_inner,
+		  y:size(3) - unstable_region_offset * scale_inner, y:size(2) - unstable_region_offset * scale_inner)
    if options.x_upsampling then
    if options.x_upsampling then
-      x = iproc.crop(x, unstable_region_offset, unstable_region_offset,
-		     x:size(3) - unstable_region_offset, x:size(2) - unstable_region_offset)
-      y = iproc.crop(y, unstable_region_offset, unstable_region_offset,
-		     y:size(3) - unstable_region_offset, y:size(2) - unstable_region_offset)
       assert(x:size(2) % 4 == 0 and x:size(3) % 4 == 0)
       assert(x:size(2) % 4 == 0 and x:size(3) % 4 == 0)
       assert(x:size(1) == y:size(1) and x:size(2) == y:size(2) and x:size(3) == y:size(3))
       assert(x:size(1) == y:size(1) and x:size(2) == y:size(2) and x:size(3) == y:size(3))
    else
    else
       assert(x:size(1) == y:size(1) and x:size(2) * scale == y:size(2) and x:size(3) * scale == y:size(3))
       assert(x:size(1) == y:size(1) and x:size(2) * scale == y:size(2) and x:size(3) * scale == y:size(3))
    end
    end
-   local scale_inner = scale
-   if options.x_upsampling then
-      scale_inner = 1
-   end
    local batch = {}
    local batch = {}
-
    for i = 1, n do
    for i = 1, n do
       local xc, yc = pairwise_utils.active_cropping(x, y,
       local xc, yc = pairwise_utils.active_cropping(x, y,
 						    size,
 						    size,

+ 0 - 4
lib/pairwise_transform_utils.lua

@@ -57,10 +57,6 @@ function pairwise_transform_utils.active_cropping(x, y, size, scale, p, tries)
       local xc = iproc.crop(x, xi, yi, xi + size / scale, yi + size / scale)
       local xc = iproc.crop(x, xi, yi, xi + size / scale, yi + size / scale)
       return xc, yc
       return xc, yc
    else
    else
-      local test_scale = 2
-      if test_scale < scale then
-	 test_scale = scale
-      end
       local lowres = gm.Image(y, "RGB", "DHW"):
       local lowres = gm.Image(y, "RGB", "DHW"):
 	    size(y:size(3) * 0.5, y:size(2) * 0.5, "Box"):
 	    size(y:size(3) * 0.5, y:size(2) * 0.5, "Box"):
 	    size(y:size(3), y:size(2), "Box"):
 	    size(y:size(3), y:size(2), "Box"):