From 6220afc31a790bab4858a10fc7736a588b5e0d6b Mon Sep 17 00:00:00 2001 From: nagadomi Date: Mon, 24 Oct 2016 17:23:09 +0900 Subject: [PATCH] Fix a bug in random negate --- lib/data_augmentation.lua | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/data_augmentation.lua b/lib/data_augmentation.lua index 3c76b5f..a8f9b60 100644 --- a/lib/data_augmentation.lua +++ b/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