Quellcode durchsuchen

Fix a bug in random negate

nagadomi vor 8 Jahren
Ursprung
Commit
6220afc31a
1 geänderte Dateien mit 3 neuen und 3 gelöschten Zeilen
  1. 3 3
      lib/data_augmentation.lua

+ 3 - 3
lib/data_augmentation.lua

@@ -123,8 +123,8 @@ end
 function data_augmentation.pairwise_negate(x, y, p)
    if torch.uniform() < p then
       assert(x:size(2) == y:size(2) and x:size(3) == y:size(3))
-      x = iproc.negate(x, r)
-      y = iproc.rotate(y, r)
+      x = iproc.negate(x)
+      y = iproc.negate(y)
       return x, y
    else
       return x, y
@@ -133,7 +133,7 @@ end
 function data_augmentation.pairwise_negate_x(x, y, p)
    if torch.uniform() < p then
       assert(x:size(2) == y:size(2) and x:size(3) == y:size(3))
-      x = iproc.negate(x, r)
+      x = iproc.negate(x)
       return x, y
    else
       return x, y