call NumpyToTensor last

ivadomed · Jun 4, 2021 · 2678b4d · 2678b4d
1 parent a1a0aaf
commit 2678b4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/ivadomed/transforms.py b/ivadomed/transforms.py
@@ -148,10 +148,10 @@ def __call__(self, sample, metadata, data_type='im'):
             return None, None
         else:
             for tr in self.transform[data_type].transforms:
-                if isinstance(tr, NormalizeInstance):
-                    numpy_to_tensor = NumpyToTensor()
-                    sample, metadata = numpy_to_tensor(sample, metadata)
                 sample, metadata = tr(sample, metadata)
+
+            numpy_to_tensor = NumpyToTensor()
+            sample, metadata = numpy_to_tensor(sample, metadata)
             return sample, metadata