Make random noise transform not in-place

fepegar · Jan 7, 2020 · bf60d5d · bf60d5d
1 parent 46d7a4d
commit bf60d5d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torchio/transforms/random_noise.py b/torchio/transforms/random_noise.py
@@ -18,7 +18,7 @@ def apply_transform(self, sample):
                 continue
             if image_dict['type'] != INTENSITY:
                 continue
-            add_noise(image_dict['data'], std)
+            image_dict['data'] = add_noise(image_dict['data'], std)
         return sample
 
     @staticmethod
@@ -29,4 +29,5 @@ def get_params(std_range):
 
 def add_noise(data, std):
     noise = torch.FloatTensor(*data.shape).normal_(mean=0, std=std)
-    data += noise
+    data = data + noise
+    return data